首页 > 其他分享 >P4516

P4516

时间:2024-10-22 18:10:00浏览次数:1  
标签:int lim void maxk maxn P4516 mod

sb大括号,卡我114514ms

#include <bits/stdc++.h>
using namespace std;
const int maxn=100005,maxk=104,mod=1000000007;
int f[maxn][maxk][2][2];
int g[maxk][2][2];
int lim[maxn];
vector<int> G[maxn];
inline void __a(int &x){ if(x >= mod) x -= mod;}
int N,K;
void dp(int u,int fa){
	f[u][0][0][0]=1;
	f[u][1][1][0]=1;
	lim[u]=1;
	for(int v:G[u]){
		if(v!=fa){
			dp(v,u);
			int nlim=min(lim[u]+lim[v],K);
			for(int a=0;a<=lim[u];a++)for(int b=0;b<=lim[v]&&a+b<=nlim;b++)for(int p1=0;p1<2;p1++)for(int q1=0;q1<2;q1++)for(int p2=0;p2<2;p2++)for(int q2=0;q2<2;q2++)if(q2|p1)__a(g[a+b][p1][q1|p2]+=1ll*f[u][a][p1][q1]*f[v][b][p2][q2]%mod);
			for(int k=0;k<=nlim;k++)for(int p=0;p<2;p++)for(int q=0;q<2;q++)f[u][k][p][q]=g[k][p][q];
			for(int k=0;k<=nlim;k++)g[k][0][0]=g[k][0][1]=g[k][1][0]=g[k][1][1]=0;
			lim[u]=nlim;
		}
	}
}
int main(){
	scanf("%d %d",&N,&K);
	for(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\n",(f[1][K][0][1]+f[1][K][1][1])%mod);
}

标签:int,lim,void,maxk,maxn,P4516,mod
From: https://www.cnblogs.com/zan-mei-tai-yang/p/18493456

相关文章