似乎和曾经一道简单树形
d
p
dp
dp类似?
f
[
i
]
[
j
]
[
0
/
1
]
f[i][j][0/1]
f[i][j][0/1]表示
i
i
i的子树点亮
j
j
j个,当前这个点点不点亮的方案
只不过这个是自己不会点亮自己
于是
f
[
i
]
[
j
]
[
0
/
1
]
[
0
/
1
]
f[i][j][0/1][0/1]
f[i][j][0/1][0/1]多一维表示自己是否被点亮
要考虑的情况似乎有点多?
卡空间毒瘤差评
#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline int read(){
char ch=getchar();
int res=0,f=1;
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=getchar();
return res*f;
}
const int N=100005,K=105;
int adj[N],siz[N],nxt[N<<1],to[N<<1],cnt,n,k;
const ll mod=1e9+7;
int f[100010][105][2][2];
ll tmp[K][2][2];
void add(int &a,ll b){
a=(a+((int)(b%mod)))%mod;
}
inline void addedge(int u,int v){
nxt[++cnt]=adj[u],adj[u]=cnt,to[cnt]=v;
}
void dfs(int u,int fa){
siz[u]=1,f[u][0][0][0]=f[u][1][1][0]=1;
for(int e=adj[u];e;e=nxt[e]){
int v=to[e];if(v==fa)continue;
dfs(v,u);
for(int i=0;i<=siz[u]&&i<=k;i++){
for(int f1=0;f1<=1;f1++)for(int f2=0;f2<=1;f2++)
tmp[i][f1][f2]=f[u][i][f1][f2],f[u][i][f1][f2]=0;
}
for(int i=0;i<=min(siz[u],k);i++){
for(int j=0;j<=min(siz[v],k-i);j++){
add(f[u][i+j][0][0],(ll)tmp[i][0][0]*(ll)f[v][j][0][1]);
add(f[u][i+j][1][0],(ll)tmp[i][1][0]*(ll)(f[v][j][0][0]+f[v][j][0][1]));
add(f[u][i+j][0][1],(ll)tmp[i][0][0]*(ll)f[v][j][1][1]+(ll)tmp[i][0][1]*(ll)(f[v][j][0][1]+f[v][j][1][1]));
add(f[u][i+j][1][1],(ll)tmp[i][1][0]*(ll)(f[v][j][1][1]+f[v][j][1][0])+(ll)tmp[i][1][1]*(ll)((ll)(f[v][j][0][0]+f[v][j][0][1])+(ll)(f[v][j][1][1]+f[v][j][1][0])));
}
}
siz[u]+=siz[v];
}
}
int main(){
n=read(),k=read();
for(int i=1;i<n;i++){
int u=read(),v=read();
addedge(u,v),addedge(v,u);
}
dfs(1,0);
cout<<((ll)f[1][k][1][1]+(ll)f[1][k][0][1])%mod<<'\n';
}