【JZOJ 杂题选讲】【CF1276D】Tree Elimination

思路

又是一个神仙dp题
考虑一个点，要么被它的父亲边染色，要么被其它边染色，要么不染色。而如果其它边的染色顺序在它的父亲边前将影响到它父亲边是否能选。
所以设f[x][0/1/2/3]表示x的子树中，x点不染色，被父亲边前的边染色，被父亲边染色，被父亲边之后的边染色的方案数。
讨论一下并转移就好了。
举个例子：f[x][1]考虑枚举一个y，那么y要么不选，要么在这条边之后选，所以贡献f[y][0/2]。然后y前的儿子一定要选，f[z][1/2]。y后的除了父亲都能选，所以贡献f[z][0/1/3]。
其他状态类似

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=200010,Mod=998244353;
int n,f[N][4],a[N],b[N],c[N];
vector<int>G[N];
void dfs(int u,int ff)
{
	if(G[u].size()==1&&u!=1){f[u][1]=f[u][3]=1;return;}
	for(int v:G[u])if(v!=ff)dfs(v,u);
	int m=0,pa=0;
	for(int v:G[u])
		if(v==ff)pa=m;
		else
		{
			a[++m]=(f[v][0]+f[v][1])%Mod;
			b[m]=(f[v][2]+f[v][3])%Mod;
			c[m]=(0ll+f[v][0]+f[v][2]+f[v][3])%Mod;
		}
	a[0]=c[m+1]=1;
	for(int i=1;i<=m;i++)a[i]=1ll*a[i]*a[i-1]%Mod;
	for(int i=m;i;i--)c[i]=1ll*c[i]*c[i+1]%Mod;
	for(int i=1;i<=m;i++)f[u][(i>pa)*2]=(f[u][(i>pa)*2]+1ll*b[i]*a[i-1]%Mod*c[i+1]%Mod)%Mod;
	f[u][1]=1ll*a[pa]*c[pa+1]%Mod;f[u][3]=a[m];
}
int main()
{
	n=gi();
	for(int i=1;i<n;i++){int u=gi(),v=gi();G[u].push_back(v);G[v].push_back(u);}
	dfs(1,1);
	printf("%d\n",((f[1][0]+f[1][2])%Mod+f[1][3])%Mod);
}