洛谷 P4516 [JSOI2018] 潜入行动

题目来源于：洛谷

题目本质：背包，树形dp

解题思路：

假设当前合并两个背包f[u][a][p1][q1] 和f[v][b][p2][q2] ，其中 v 是 u 的儿子。考虑合并后的f[u][a+b][p3][q3],q3 是合并后点 u 是否被监听，有两种情况：u 之前已经被监听，u 现在被 v 监听。即：q3=q1∣p2 。

完整代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=100005;
const int mod=1000000007;
int N,K;
int f[MAXN][105][2][2];
int g[105][2][2];
int size[MAXN];
vector<int> G[MAXN];
int Mod(ll x,ll y){
    x%=mod,y%=mod;
    return (int)(x+y)%mod;
}
void dp(int u,int fa){
    size[u]=1;
    f[u][0][0][0]=f[u][1][1][0]=1;
    for(vector<int>::iterator it=G[u].begin();it!=G[u].end();it++){
        int v=*it;
        if(v==fa) continue;
        dp(v,u);
        for(register int i=0;i<=min(size[u],K);++i){
            g[i][0][0]=f[u][i][0][0],f[u][i][0][0]=0;
            g[i][0][1]=f[u][i][0][1],f[u][i][0][1]=0;
            g[i][1][0]=f[u][i][1][0],f[u][i][1][0]=0;
            g[i][1][1]=f[u][i][1][1],f[u][i][1][1]=0;
        }
        for(register int i=0;i<=min(size[u],K);++i){
            for(register int j=0;j<=min(size[v],K-i);++j){
                f[u][i+j][0][0]=Mod((ll)f[u][i+j][0][0],(ll)g[i][0][0]*(ll)f[v][j][0][1]);
                f[u][i+j][0][1]=Mod((ll)f[u][i+j][0][1],(ll)g[i][0][0]*(ll)f[v][j][1][1]+(ll)g[i][0][1]*((ll)f[v][j][1][1]+(ll)f[v][j][0][1]));
                f[u][i+j][1][0]=Mod((ll)f[u][i+j][1][0],(ll)g[i][1][0]*((ll)f[v][j][0][0]+(ll)f[v][j][0][1]));
                f[u][i+j][1][1]=Mod((ll)f[u][i+j][1][1],(ll)g[i][1][0]*((ll)f[v][j][1][0]+(ll)f[v][j][1][1])+(ll)g[i][1][1]*((ll)f[v][j][0][0]+(ll)f[v][j][0][1]+(ll)f[v][j][1][0]+(ll)f[v][j][1][1]));
            }
        }
        size[u]+=size[v];
    }
}
int main(){
    scanf("%d%d",&N,&K);
    for(register int i=1;i<N;++i){
        int u,v;
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dp(1,0);
    printf("%d",(int)(f[1][K][0][1]+f[1][K][1][1])%mod);
    return 0;
}