NOI Online #2 提高组 第三题：游戏

前言

这道题上来一看就是树的题目。

题目都告诉我了

但是我还爆零了。

---

题目

传送门

解析

我们记 “恰好有$k$次非平局” 的方案数是$g(k)$，“钦定有 $k$次非平局” 的方案数是$f(k)$ 。发现这是一个二项式反演的形式，由二项式反演有公式：

$$f(n)=\sum _{i=n}^m\left(\genfrac{}{}{0ex}{}{i}{n}\right)g(i)\iff g(n)=\sum _{i=n}^m(-1{)}^{i-n}(\genfrac{}{}{0ex}{}{i}{n})f(i)$$

所以我们只要求出$f(k)$就做完了这个题。我们可以$dp$求出$f$，设 $dp[u][x]$ 表示 $u$ 子树内钦定选择了 $x$ 对点，这$x$对点均呈 祖先 - 子孙 的关系（也就是这 $x$ 局钦定会比出胜负）。于是发现这东西可以直接树形背包算。

转移就是正常的树形背包，最后还需要考虑钦定这个点和某个后代的情况，这种情况的转移就是从子树内未被钦定的点中选择一个点。

(注：钦定有 k 次就是有某 k 个匹配必然被选，剩下的可以选也可以不选，恰好 k 次是必须正好 k 个匹配）

代码

#include<bits/stdc++.h>
using namespace std;
const int Maxn=5006;
#define rep(i, a, b) for (int i=(a),i##end=(b);i<=i##end;++i) //将for重新定义，我一会儿会有博客来解释
#define per(i, a, b) for (int i=(a),i##end=(b);i>=i##end;--i)
const int Mod=998244353;
int n,Ar[Maxn];
int head[Maxn],nxt[Maxn<<1],to[Maxn<<1],cedge=0;
void adde(int u,int v)
{
	to[++cedge]=v;
	nxt[cedge]=head[u];
	head[u]=cedge;
}
int dp[Maxn][Maxn],Size[Maxn],sz[Maxn],pd[Maxn];
void dfs(int u,int fa)
{
	Size[u]=1;
	sz[u]=Ar[u];
	dp[u][0]=1;
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v==fa) continue;
		dfs(v,u);
		rep(k,0,Size[u]+Size[v]) pd[k]=0;
		rep(j,0,min(Size[u],n/2))
		{
			if(dp[u][j])
			{
				rep(k,0,min(Size[v],n/2-j))
				{
					if(dp[v][k])
					{
						pd[j+k]=(pd[j+k]+dp[u][j]*1ll*dp[v][k]%Mod)%Mod;
					}
				}
			}
		}
		rep(k,0,Size[u]+Size[v]) dp[u][k]=pd[k];
        Size[u]+=Size[v];
		sz[u]+=sz[v];
	}
	per(i,min(sz[u],Size[u]-sz[u]),1)
	{
		dp[u][i]=(dp[u][i]+dp[u][i-1]*1ll*((Ar[u]?(Size[u]-sz[u]):(sz[u]))-(i-1))%Mod)%Mod;
	}
}
int J[Maxn],iJ[Maxn];
int Quick_Pow(int x,int a)
{
    int ret=1;
    while(a)
	{
        if(a&1) ret=1ll*ret*x%Mod;
        x=1ll*x*x%Mod;
		a>>= 1;
    }
    return ret;
}

int C(int x,int y)
{
    return J[x]*1ll*iJ[y]%Mod*iJ[x-y]%Mod;
}

int main()
{
	scanf("%d",&n);
    J[0]=iJ[0]=1;
    rep(i,1,Maxn-1) J[i]=J[i-1]*1ll*i%Mod,iJ[i]=Quick_Pow(J[i],Mod-2);
    rep(i,1,n)
        scanf("%1d",Ar+i);
    int u,v;
    rep(i,2,n) scanf("%d%d",&u,&v),adde(u,v),adde(v,u);
    dfs(1,1);
    int re=0;
    rep(i,0,n/2+1) dp[1][i]=dp[1][i]*1ll*J[n/2-i]%Mod;
    rep(i,0,n/2)
	{
        re=0;
        rep(j,i,n/2)
            re+=C(j,i)*1ll*((j-i&1)?Mod-1:1)%Mod*dp[1][j]%Mod,re%=Mod;
        printf("%d\n",re);
    }
	return 0;
}