正題

nowcoder 1103-B

---

題目大意

給你一棵樹，讓你找兩條不交的鏈，長度分別為a,b，問有多少中方案

---

解題思路

設$f_{x,0～3,y}$為在x的子樹中，第1/2條鏈連了y條邊，另一條鏈連完了/沒連

然后對于每個兒子考慮作為什么計算，直接DP

---

code

#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define ll long long #define N 3010 using namespace std; ll n,a,b,x,y,tot,f[N][4][N],h[N]; struct rec {ll to,nx; }e[N<<1]; void add(ll x,ll y) {e[++tot].to=y;e[tot].nx=h[x];h[x]=tot; } void dfs(ll x,ll fa) {f[x][0][0]=1;f[x][1][0]=1;ll num,num1=0,num2=0;for(ll i=h[x];i;i=e[i].nx){ll y=e[i].to;if(y==fa)continue;dfs(y,x);num=0;for(ll j=1;j<=a;++j){if(j<a){//由兩個兒子直接構成鏈f[x][0][a]+=f[x][0][j]*f[y][0][a-j-1];num+=f[x][0][j]*f[y][0][a-j-1];f[x][2][a]+=f[x][2][j]*f[y][0][a-j-1]+f[x][0][j]*f[y][2][a-j-1];}f[x][0][j]+=f[y][0][j-1];//練到父親f[x][2][j]+=f[y][2][j-1]+f[y][0][j-1]*num2;//之前可以構成另一條鏈的}f[x][2][0]+=f[y][1][b];for(ll j=1;j<a;++j)f[x][2][j]+=(f[x][0][j]-f[y][0][j-1])*f[y][1][b];//同一條鏈只能使用一次f[x][2][a]+=(f[x][0][a]-f[y][0][a-1]-num)*f[y][1][b];num=0;for(ll j=1;j<=b;++j){if(j<b){f[x][1][b]+=f[x][1][j]*f[y][1][b-j-1];num+=f[x][1][j]*f[y][1][b-j-1];f[x][3][b]+=f[x][3][j]*f[y][1][b-j-1]+f[x][1][j]*f[y][3][b-j-1];}f[x][1][j]+=f[y][1][j-1];f[x][3][j]+=f[y][3][j-1]+f[y][1][j-1]*num1;}f[x][3][0]+=f[y][0][a];for(ll j=1;j<b;++j)f[x][3][j]+=(f[x][1][j]-f[y][1][j-1])*f[y][0][a];f[x][3][b]+=(f[x][1][b]-f[y][1][b-1]-num)*f[y][0][a];f[x][0][a]+=f[y][0][a];num1+=f[y][0][a];f[x][1][b]+=f[y][1][b];num2+=f[y][1][b];f[x][2][a]+=f[y][2][a];f[x][3][b]+=f[y][3][b];} } int main() {scanf("%lld%lld%lld",&n,&a,&b);for(ll i=1;i<n;++i){scanf("%lld%lld",&x,&y);add(x,y);add(y,x);}dfs(1,0);printf("%lld",(f[1][2][a]+f[1][3][b])*4);return 0; }

總結

以上是生活随笔為你收集整理的【树形DP】路径计数机（nowcoder 1103-B）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。