[JSOI2018]潜入行动

这道题实际上一入眼便是树形dp

我们考虑一下f[i][j][0/1][0/1]表示考虑i的子树已经放置了j个监听设备，当前节点（放设备或不放设备），当前节点（被监听或不被监听）的方案数

然后再推一下就比较显然，关键其实在你仔不仔细，耐不耐的下心来，推导过程就由读者自行解决

但是我们发现这道题这么做的话似乎要nk^2,但是由于树上dp的一些独特性质,实际上可以到nk，我认为这个优化才是本题的最大难点，作者到目前为止还不会其证明方法，具体优化过程看代码

如有大佬会上述优化的时间复杂度，请私聊我谢谢

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=1e9+7;
const int MAXN=1e5+5;
int n,k,u,v;
int f[MAXN][101][2][2],size[MAXN];
vector<int>e[MAXN];
void la(int &x,int y)
{
   x=(x+y)%mod;
}
ll mul(ll x,ll y){
   return (x*y)%mod;
}
ll p(ll x,ll y)
{
    return (x+y)%mod;
}
void dfs(int now,int fa)
{
    f[now][0][0][0]=f[now][1][1][0]=1;
    size[now]=1;
    for (int v=0;v<e[now].size();v++)
    {
        int u=e[now][v];
        if (u==fa) continue;
        dfs(u,now);
        for (int i=0;i<=min(size[now]+size[u],k);i++) 
        {
            f[0][i][0][0]=f[0][i][0][1]=f[0][i][1][0]=f[0][i][1][1]=0;
        }
        for (int i=0;i<=min(size[now],k);i++)
           for (int j=0;j<=min(size[u],k-i);j++)
           {
               la(f[0][i+j][0][0],mul(f[now][i][0][0],f[u][j][0][1]));
               la(f[0][i+j][0][1],p(mul(f[now][i][0][0],f[u][j][1][1]),mul(f[now][i][0][1],p(f[u][j][1][1],f[u][j][0][1]))));
               la(f[0][i+j][1][0],mul(f[now][i][1][0],p(f[u][j][0][1],f[u][j][0][0])));
               la(f[0][i+j][1][1],p(mul(f[now][i][1][0],p(f[u][j][1][0],f[u][j][1][1])),mul(f[now][i][1][1],p(p(f[u][j][1][0],f[u][j][1][1]),p(f[u][j][0][1],f[u][j][0][0])))));
           }
        size[now]+=size[u];   
        for (int i=0;i<=min(size[now],k);i++)
        {
            f[now][i][0][0]=f[0][i][0][0];
            f[now][i][0][1]=f[0][i][0][1];
            f[now][i][1][0]=f[0][i][1][0]; 
            f[now][i][1][1]=f[0][i][1][1];
        }   
    }
}
int main()
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        e[u].push_back(v); e[v].push_back(u);
    }
    dfs(1,0);
    // cout << f[3][2][1][1] << endl;
    printf("%d\n",(f[1][k][1][1]+f[1][k][0][1])%mod);
}