潜入行动
很容易想出一个状态,设
然后你去推,就会发现我们还需要知道根节点是否被监听这一个信息(最开始
所以我们设成
写下状态转移方程,有点复杂,与题解对照一下
update 2024.6.30
像这种一个点的状况可以影响儿子和父亲的,一般有两种设法,第一种就是像上面说的这样不考虑父亲和儿子,只考虑自己是否被影响到(无论是父亲影响的还是儿子影响的)
第二种就是像“警卫安排”一样,把父亲或者儿子的状态可考虑进来
设
具体见如下代码(temp数组是用来减掉所有儿子都不放监听器的情况的),有推导方程,也有很多细节
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e5+10,K=110;
const ll p=1e9+7;
int n,k;
int End[N<<1],Next[N<<1],Last[N];
int son[N];
int f[K][2][2],g[N][K][2][2],temp[N][K];
int read()
{
int x=0,f=1;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-f;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-48;s=getchar();}
return x*f;
}
int cnt;
void add(int x,int y)
{
End[++cnt]=y,Next[cnt]=Last[x],Last[x]=cnt;
}
void dfs(int x,int fa)
{
son[x]=1;
g[x][0][0][1]=g[x][1][1][1]=temp[x][0]=1;//初始化,只考虑x这一个孤立结点
for(int i=Last[x];i;i=Next[i])
{
int u=End[i];
if(u==fa) continue;
dfs(u,x);
for(int j=0;j<=min(k,son[x]+son[u]);j++)
for(int l=0;l<=1;l++)
for(int p=0;p<=1;p++)
f[j][l][p]=0;//f是要用来记录累加和的
for(int j=min(son[x]+son[u],k);j>=0;j--)
{
for(int K=max(0,j-son[x]);K<=min(son[u],j);K++)
//注意这里的K的起点一定要这么优化,不然会被卡常
{
f[j][0][1]=((ll)f[j][0][1]+(ll)g[x][j-K][0][1]*(g[u][K][0][0]+g[u][K][1][0])%p)%p;
f[j][1][1]=((ll)f[j][1][1]+(ll)g[x][j-K][1][1]*(g[u][K][0][1]+g[u][K][1][1])%p)%p;
}
}
son[x]+=son[u];
for(int j=0;j<=min(k,son[x]);j++)
for(int l=0;l<=1;l++)
g[x][j][l][1]=f[j][l][1];
}
//之所以要像这么拆开求,是因为要节省空间,否则的话temp数组就要多开一维变成temp[N][K][2],就会MLE了
//MLE的代码可以看hydroOJ的提交
son[x]=1;
for(int i=Last[x];i;i=Next[i])
{
int u=End[i];
if(u==fa) continue;
for(int j=min(son[x]+son[u],k);j>=0;j--)
{
ll t=0;//t是用来记录累加和的
for(int K=max(0,j-son[x]);K<=min(son[u],j);K++)
t=(t+(ll)temp[x][j-K]*g[u][K][0][0]%p)%p;
temp[x][j]=t;
}
son[x]+=son[u];
for(int j=0;j<=min(k,son[x]);j++)
for(int l=0;l<=0;l++)
g[x][j][l][0]=((ll)f[j][l][1]-temp[x][j]+p)%p;//容斥原理
}
for(int i=0;i<=k;i++) temp[x][i]=0;
temp[x][1]=1;
son[x]=1;
for(int i=Last[x];i;i=Next[i])
{
int u=End[i];
if(u==fa) continue;
for(int j=min(son[x]+son[u],k);j>=0;j--)
{
ll t=0;
for(int K=max(0,j-son[x]);K<=min(son[u],j);K++)
t=(t+(ll)temp[x][j-K]*g[u][K][0][1]%p)%p;
temp[x][j]=t;
}
son[x]+=son[u];
for(int j=0;j<=min(k,son[x]);j++)
for(int l=1;l<=1;l++)
g[x][j][l][0]=((ll)f[j][l][1]-temp[x][j]+p)%p;
}
}
int main()
{
n=read(),k=read();
for(int i=1,u,v;i<n;i++)
{
u=read(),v=read();
add(u,v),add(v,u);
}
dfs(1,0);
printf("%lld",(g[1][k][0][0]+g[1][k][1][0])%p);//最终输出答案根节点一定只能被儿子监视不能被父亲监视,因为没有父亲
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 【译】Visual Studio 中新的强大生产力特性
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 【设计模式】告别冗长if-else语句:使用策略模式优化代码结构