答案中每个岛上一定是一坨森林。岛之间的边可以有可以没有。
不过要减去所有岛都1和a[i]连通并且所有岛之间的边都存在的情况。
设f[x]表示对于一个城镇数为x的岛,连成森林的方案数。
那么
f[x]=∑i=0x−1f[i]∗v[x−i]∗Cx−i−1x−1
=(x−1)!∑i=0x−1f[i]i!∗v[x−i](x−i−1)!
其中v[i]表示i个点的生成树数量,由prufer编码得
v[i]=ii−2
然后这个东西可以分治FFT求。
设g[x]表示对于一个城镇数为x的岛,连成森林并且第一个城镇和第x个城镇连通的方案数。
那么
g[x]=∑i=2xCi−2x−2v[i]∗f[x−i]
同理可以化成卷积的形式。
然后最终答案是
2n∏f[a[i]]−∏g[a[i]]
#include <bits/stdc++.h>
using namespace std;
#define N 110000
#define ll long long
#define mod 998244353
int T,n,a1,a2;
int jc[N],njc[N],f[N],g[N],v[N];
int a[N<<2],b[N<<2],c[N<<2];
int qpow(int x,int y)
{
int ret=1;
while(y)
{
if(y&1)ret=(ll)ret*x%mod;
x=(ll)x*x%mod;y>>=1;
}
return ret;
}
void NTT(int *a,int len,int type)
{
for(int i=0,t=0;i<len;i++)
{
if(i<t)swap(a[i],a[t]);
for(int j=len>>1;(t^=j)<j;j>>=1);
}
for(int i=2;i<=len;i<<=1)
{
int wn=qpow(3,(mod-1)/i);
for(int j=0;j<len;j+=i)
{
int w=1,t;
for(int k=0;k<i>>1;k++,w=(ll)w*wn%mod)
{
t=(ll)w*a[j+k+(i>>1)]%mod;
a[j+k+(i>>1)]=(a[j+k]-t+mod)%mod;
a[j+k]=(a[j+k]+t)%mod;
}
}
}
if(type==-1)
{
for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);
int t=qpow(len,mod-2);
for(int i=0;i<len;i++)a[i]=(ll)a[i]*t%mod;
}
}
void update(int l,int r,int mid)
{
int len=1;
for(;len<(r-l+1)<<1;len<<=1);
for(int i=0;i<len;i++)a[i]=b[i]=0;
for(int i=l;i<=mid;i++)a[i-l]=(ll)f[i]*njc[i]%mod;
for(int i=1;i<=r-l;i++)b[i-1]=(ll)v[i]*njc[i-1]%mod;
NTT(a,len,1);NTT(b,len,1);
for(int i=0;i<len;i++)c[i]=(ll)a[i]*b[i]%mod;
NTT(c,len,-1);
for(int i=mid+1;i<=r;i++)
f[i]=(f[i]+c[i-l-1])%mod;
}
void solve(int l,int r)
{
if(l==r)
{
if(l==1)f[l]=1;
else f[l]=((ll)f[l]*jc[l-1]%mod+v[l])%mod;
return;
}
int mid=(l+r)>>1;
solve(l,mid);
update(l,r,mid);
solve(mid+1,r);
}
int main()
{
//freopen("tt.in","r",stdin);
scanf("%d",&T);
n=100000;jc[0]=njc[0]=1;v[1]=1;
for(int i=2;i<=n;i++)v[i]=qpow(i,i-2);
for(int i=1;i<=n;i++)jc[i]=(ll)jc[i-1]*i%mod;
njc[n]=qpow(jc[n],mod-2);
for(int i=n-1;i>=1;i--)njc[i]=(ll)njc[i+1]*(i+1)%mod;
solve(1,n);f[0]=1;
int len=1;for(;len<n<<1;len<<=1);g[1]=1;
for(int i=0;i<len;i++)a[i]=b[i]=0;
for(int i=2;i<=n;i++)a[i]=(ll)v[i]*njc[i-2]%mod;
for(int i=0;i<=n-2;i++)b[i]=(ll)f[i]*njc[i]%mod;
NTT(a,len,1);NTT(b,len,1);
for(int i=0;i<len;i++)c[i]=(ll)a[i]*b[i]%mod;
NTT(c,len,-1);
for(int i=2;i<=n;i++)g[i]=(ll)c[i]*jc[i-2]%mod;
while(T--)
{
scanf("%d",&n);a1=a2=1;
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++)a1=(ll)a1*f[a[i]]%mod;
a1=(ll)a1*qpow(2,n)%mod;
for(int i=1;i<=n;i++)a2=(ll)a2*g[a[i]]%mod;
printf("%d\n",(a1-a2+mod)%mod);
}
return 0;
}