【TEST190403】分治NTT & SAM+树剖 & 高斯消元+概率DP

game

枚举被所有人经过的路径$(u,v)$(显然只有一条连续路径被所有人经过)。

考虑$LCA(u,v)$不为$u$或$v$时，$f_u,f_v$分别表示在$u,v$子树中给$K$个人安排出起始点(可重合)的方案数。

发现$f_x$就是生成函数$K!(\sum _{i=0}^{\infty }\frac{1}{i!})\prod _{y\in so{n}_x}(1+s{z}_y\cdot x)$的$x^K$项的系数。

分治NTT求出$f_x$的复杂度是$degre{e}_x{\log }^{2}degre{e}_x$的，故总复杂度为$n{\log }^{2}n$。

$dfs$一遍处理出不为祖先关系的点对答案。

对于$LCA(u,v)=u$的情况，设$v\in x,x\in so{n}_u$则$f_u$需要乘上$\frac{1+(n-s{z}_u)x}{1+s{z}_x}$，多项式乘或除另一个一次多项式的复杂度显然是线性的，而$n$个点的树中$s{z}_x$的值最多只有$\sqrt{n}$种。复杂度是$n\sqrt{n}$的。

总复杂度$O(n\sqrt{n}+n{\log }^{2}n)$

#include<bits/stdc++.h>
#define pb push_back
#define fi first
#define sc second
using namespace std;
typedef long long ll;
const int N=1e5+100;
const int mod=998244353,g=3,nvg=332748118;

int num,n,m,ans,frac,nv[N];
int rep[N<<3],sz[N],sum[N],val[N];
int head[N],to[N<<1],nxt[N<<1],tot;
vector<int>xs[N];
map<int,int>mp[N];

char cp;
inline void rd(int &x)
{
   
	cp=getchar();x=0;
	for(;!isdigit(cp);cp=getchar());
	for(;isdigit(cp);cp=getchar()) x=x*10+(cp^48);
}

inline int fp(int x,int y)
{
   
	int re=1;
	for(;y;y>>=1,x=(ll)x*x%mod)
	  if(y&1) re=(ll)re*x%mod;
	return re