LG P5206 [WC2019] 数树

题目
问题0:
设存在$l$条边同时存在。
答案为 $F_0(l)=y^{n-l}$

问题1:
设对于红树S有$G(S)$条边与蓝树同时存在：
答案为$F_1={\sum }_S{F}_0(G(S))$
G不方便求。
设红树边集为$A$，蓝树边集为$B$
设$C(S)={\sum }_A[S\subset (A\cap B)]$即红蓝树交集至少包含S的边的方案数。
答案为${\sum }_S{F}_0(|S|){\sum }_{S\subset T}(-1{)}^{|T|-|S|}C(T)$
和式变换一下得到：
${\sum }_T(-1{)}^{|T|}C(T){\sum }_{S\subset T}{F}_0(|S|)(-1{)}^{|S|}$
${\sum }_T(-1{)}^{|T|}C(T){\sum }_{i=0}^{|T|}{y}^{n-i}(-1{)}^i\left(\genfrac{}{}{0ex}{}{i}{|T|}\right)$
$y^n{\sum }_T(-1{)}^{|T|}C(T){\sum }_{i=0}^{|T|}(-y{)}^{-i}\left(\genfrac{}{}{0ex}{}{i}{|T|}\right)$
后半部分还可以反用二项式定理化简
$y^n{\sum }_T(-1{)}^{|T|}C(T)(1-\frac{1}{y}{)}^{|T|}$
$y^n{\sum }_{T}C(T)(\frac{1}{y}-1{)}^{|T|}$
这样胡乱操作一波发现我们的算法还是指数级的。
但是发现C(T)是可计算（？？）的,因为至少包含可以让我们不用管蓝树。
设边集$T$将原图分成了$k$个联通块，第i个的大小为$a_k$
（容易）猜想到$C(T)=n^{k-2}{\prod }_i{a}_i$
这个从意义上很好证明。
类似于Cayley定理的非prufer序列及Matrix-Tree证明
朴素的证明更能让人感到踏实。

我们先做有根树的，一次加入k-1条边的（所以最后要除上$(k-1)!$）生成树。
每次加入一条边，我们枚举这条边的父亲方向的点（因为有根所以有向）(一共有n种方案)，
然后再枚举儿子方向的联通块（第i次有k-i种方案)，
除上$(k-1)!$和$n$，乘上每个联通块选一个往父亲方向连${\prod }_i{a}_i$。
就是这个了。注意不是$k^{k-2}{\prod }_{i}ai$

然后我们可以继续化式子。
$y^n{\sum }_T{n}^{n-|T|-2}(\frac{1}{y}-1{)}^{|T|}{\prod }_i^{n-|T|}{a}_i$
$y^n(\frac{1}{y}-1{)}^n{n}^{-2}{\sum }_T{\prod }_i^{n-|T|}{a}_i\ast \frac{n}{\frac{1}{y}-1}$
为啥要这么做呢？因为后面的sigma可以$O(n)dp$
好像是可以吧。。。
具体看我的另一篇博客
注意这个$T需要\subset B$所以这个问和第一颗树还是有关系的。

问题3：
一开始的式子是没多少变化的：$F_2={\sum }_S{F}_0(S){\sum }_{S\subset T}(-1{)}^{|T|-|S|}{C}^2(T)$
所以在C(T)拆开之前的都可以把$C(T)$换为$C^2(T)$
$F_2=y^n{\sum }_T{C}^2(T)(\frac{1}{y}-1{)}^{|T|}$
$F_2=y^n(\frac{1}{y}-1{)}^n{n}^{-4}{\sum }_T{\prod }_i^{n-|T|}{a}_i^2\ast \frac{n^2}{(\frac{1}{y}-1)}$
这里的$T$是所有不成环的边集了。
这个题要求不一定联通，可以求出一定联通的指数生成函数（$f(x)={\sum }_{i=0}^n{i}^2\ast \frac{n^2}{\frac{1}{y}-1}\ast i^{i-2}\frac{x^i}{i!}$），然后求一个$exp$即可。

AC Code:

#include<bits/stdc++.h>
#define maxn 300005
#define mod 998244353
using namespace std;

int wlen,w[maxn]={1},r[maxn],lg[maxn],inv[maxn]={1,1};
int Pow(int base,int k){
	int ret=1;
	for(;k;k>>=1,base=1ll*base*base%mod)
		if(k&1)
			ret=1ll*ret*base%mod;
	return ret;
}
void Init(int n){
	for(wlen=1;n>=(2*wlen);wlen<<=1);
	for(int i=1,pw=Pow(3,(mod-1)/(2*wlen));i<=2*wlen;i++) w[i] = 1ll * w[i-1] * pw % mod;
	for(int i=2;i<=2*wlen;i++) lg[i]=lg[i>>1]+1,inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
}
void NTT(int *A,int n,int typ){
	int lgn = lg[n];
	for(int i=1;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(lgn-1));
	for(int i=1;i<n;i++) if(i<r[i]) swap(A[i],A[r[i]]);
	for(int L=2;L<=n;L<<=1)
		for(int st=0,l=L>>1,inc=wlen/l;st<n;st+=L)
			for(int k=st,x=0;k<st+l;k++,x+=inc){
				int tmp = 1ll * (typ == 1 ? w[x] : w[2*wlen-x]) * A[k+l] % mod;
				A[k+l] = (A[k] - tmp) % mod , A[k] = (A[k] + tmp) % mod;
			}
	if(typ == -1)
		for(int i=0,invn=Pow(n,mod-2);i<n;i++)
			A[i] = 1ll * A[i] * invn % mod;
}
void INV(int *A,int *B,int n){
	B[0]=Pow(A[0],mod-2),B[1]=0;
	static int TMP[maxn];
	for(int k=2;k<(n<<1);k<<=1){
		for(int i=0;i<(k<<1);i++) TMP[i]=i<k?A[i]:0,B[i]=i<k?B[i]:0;
		NTT(B,k<<1,1),NTT(TMP,k<<1,1);
		for(int i=0;i<(k<<1);i++) B[i]=1ll*B[i]*(2-1ll*B[i]*TMP[i]%mod)%mod;
		NTT(B,k<<1,-1);
		for(int i=min(n,k);i<(k<<1);i++) B[i] = 0;
	}
}
void CLN(int *A,int *B,int n){
	static int TMP[maxn];
	INV(A,B,n);
	int len = 1<<(lg[2*n-2]+1);
	for(int i=0;i<len;i++) TMP[i]=i<n-1?A[i+1]*1ll*(i+1)%mod:0,B[i]=i<n?B[i]:0;
	NTT(TMP,len,1),NTT(B,len,1);
	for(int i=0;i<len;i++) TMP[i]=1ll*TMP[i]*B[i]%mod;
	NTT(TMP,len,-1);
	for(int i=0;i<len;i++) B[i]=(i&&i<n)?1ll*TMP[i-1]*inv[i]%mod:0;
}
void EXP(int *A,int *B,int n){
	static int TMP[maxn];
	B[0]=1,B[1]=0;
	for(int k=2;k<(n<<1);k<<=1){
		CLN(B,TMP,k);
		for(int i=0;i<(k<<1);i++) TMP[i]=i<k?((i==0)-TMP[i]+A[i])%mod:0,B[i]=i<k?B[i]:0;
		NTT(TMP,k<<1,1),NTT(B,k<<1,1);
		for(int i=0;i<(k<<1);i++) B[i]=1ll*B[i]*TMP[i]%mod;
		NTT(B,k<<1,-1);
		for(int i=min(k,n);i<(k<<1);i++) B[i]=0;
	}
}

int n,y,op,K;

namespace SOLVE0{
	map<pair<int,int>,int>mp;
	void solve(){
		int u,v,ans=n;
		for(int i=1;i<n;i++){
			scanf("%d%d",&u,&v);
			if(u>v) swap(u,v);
			mp[make_pair(u,v)]=1;
		}
		for(int i=1;i<n;i++){
			scanf("%d%d",&u,&v);
			if(u>v) swap(u,v);
			ans-=mp.count(make_pair(u,v));
		}
		printf("%d\n",Pow(y,ans));
	}
}

namespace SOLVE1{
	int f[maxn]={},g[maxn]={};
	int info[maxn]={},Prev[maxn<<1]={},to[maxn<<1]={},cnt_e=0;
	void Node(int u,int v){ Prev[++cnt_e]=info[u],info[u]=cnt_e,to[cnt_e]=v; }
	void dfs(int now,int ff){
		f[now]=1,g[now]=K;
		for(int i=info[now];i;i=Prev[i])
			if(to[i]!=ff){
				dfs(to[i],now);
				g[now]=(1ll*g[now]*g[to[i]]+1ll*g[now]*f[to[i]]+1ll*f[now]*g[to[i]])%mod;
				f[now]=(1ll*f[now]*f[to[i]]+1ll*f[now]*g[to[i]])%mod;
			}
	}
	void solve(){
		if(y==1){ printf("%d\n",Pow(n,n-2));return; }
		for(int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),Node(u,v),Node(v,u);
		dfs(1,0);
		g[1]=1ll*g[1]*Pow(y,n)%mod*Pow(Pow(y,mod-2)-1,n)%mod*Pow(n,mod-3)%mod;
		printf("%d\n",(g[1]+mod)%mod);
	}
}

namespace SOLVE2{
	int A[maxn],B[maxn],Fac[maxn]={1,1},iFac[maxn]={1,1};
	void solve(){
		if(y==1){ printf("%d\n",Pow(n,2*n-4));return; }
		Init(2*(n+1));K=1ll*K*n%mod;
		for(int i=2;i<=n;i++) Fac[i]=1ll*Fac[i-1]*i%mod,iFac[i]=1ll*iFac[i-1]*inv[i]%mod;
		for(int i=1;i<=n;i++) A[i]=1ll*Pow(i,i)*K%mod*iFac[i]%mod;
		EXP(A,B,n+1);
		B[n] = 1ll * B[n] * Fac[n] % mod;
		B[n] = 1ll * B[n] * Pow(1-y,n) % mod * Pow(n,mod-5) % mod;
		printf("%d\n",(B[n]+mod)%mod);
	}
}

int main(){
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	scanf("%d%d%d",&n,&y,&op);
	K=1ll*n*Pow(Pow(y,mod-2)-1,mod-2)%mod;
	if(op==0) SOLVE0::solve();
	if(op==1) SOLVE1::solve();
	if(op==2) SOLVE2::solve();
}