BZOJ5314: [Jsoi2018]潜入行动

BZOJ5314: [Jsoi2018]潜入行动

https://lydsy.com/JudgeOnline/problem.php?id=5314

分析:

• 裸树形背包，设\(f[x][i][0/1][0/1]\)表示\(x\)子树选了\(i\)个点，是否能覆盖\(x\)，是否选了\(x\)。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
using namespace std;
#define N 100050
typedef long long ll;
#define mod 1000000007
int head[N],to[N<<1],nxt[N<<1],cnt,n,K,siz[N];
int f[N][105][2][2],tmp[105][2][2];
inline void add(int u,int v) {
	to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
}
void upd(int &x,int y) {
	x+=y; if(x>=mod) x-=mod;
}
void dfs(int x,int y) {
	int i,j,k,p1,p2,q1,q2;
	siz[x]=1;
	f[x][0][0][0]=1;
	f[x][1][0][1]=1;
	for(i=head[x];i;i=nxt[i]) if(to[i]!=y) {
		dfs(to[i],x);
		int t=to[i];
		for(j=0;j<=siz[x]&&j<=K;j++) {
			for(p1=0;p1<2;p1++)for(q1=0;q1<2;q1++) {
				tmp[j][p1][q1]=f[x][j][p1][q1];
				f[x][j][p1][q1]=0;
			}
		}
		for(j=0;j<=siz[x]&&j<=K;j++) {
			for(k=0;k<=siz[t]&&j+k<=K;k++) {
				for(p1=0;p1<2;p1++) {
					for(q1=0;q1<2;q1++) if(tmp[j][p1][q1]) {
						for(p2=!q1;p2<2;p2++)  {
							for(q2=0;q2<2;q2++) {
								upd(f[x][j+k][p1|q2][q1],ll(tmp[j][p1][q1])*f[t][k][p2][q2]%mod);
							}
						}
					}
				}
			}
		}
		siz[x]+=siz[to[i]];
	}
}
int main() {
	scanf("%d%d",&n,&K);
	int i,x,y;
	for(i=1;i<n;i++) {
		scanf("%d%d",&x,&y);
		add(x,y); add(y,x);
	}
	dfs(1,0);
	printf("%d\n",(f[1][K][1][0]+f[1][K][1][1])%mod);
}
/*
5 3
1 2
2 3
3 4
4 5
*/