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
#define xx first
#define yy second
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair pii;
const int MAXN=3e5+5;
const int MOD=998244353;
vector 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