Codeforces Round #522 (Div. 1) C. Vasya and Maximum Matching（dp）

题目链接：http://codeforces.com/contest/1078/problem/C

 

dp[x][0]表示当前节点没有被匹配，且在后续可被匹配，可不被匹配的方案数

dp[x][1]表示当前节点没有被匹配，在后续当中必须被匹配的方案数

dp[x][2]表示当前节点已经被匹配的方案数

 

注意：

如果一个儿子v已经被匹配，则当前节点可以向v连边，状态从0->1或者从2->2

 

代码：

#include<bits/stdc++.h>
#define xx first
#define yy second
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int MAXN=3e5+5;
const int MOD=998244353;
vector<int> E[MAXN];
int dp[MAXN][3];
int g[3];
void dfs(int now,int fa=0)
{
	dp[now][0]=1;
	for(int v:E[now])
	{
		if(v==fa) continue;
		dfs(v,now);
		g[0]=1LL*dp[now][0]*(dp[v][0]+dp[v][2])%MOD;
		g[1]=(1LL*dp[now][0]*dp[v][2]+1LL*dp[now][1]*(dp[v][0]+2LL*dp[v][2]))%MOD;
		g[2]=(1LL*(dp[now][0]+dp[now][1])*(dp[v][0]+dp[v][1])+dp[now][2]*(dp[v][0]+2LL*dp[v][2]))%MOD;
		memcpy(dp[now],g,sizeof(g));
	}
}
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int n;
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		E[u].pb(v);
		E[v].pb(u);
	}
	dfs(1);
	printf("%d\n",(dp[1][0]+dp[1][2])%MOD);
	return 0;
}