洛谷 P4221 [WC2018]州区划分 fwt

题目：
https://www.luogu.org/problemnew/show/P4221

分析：
设$g[s]$表示选取子集为$s$的分子的贡献（暂时不考虑分母）。
显然存在欧拉回路是为0，否则为所有包含点权值和的$p$次方。注意当图不连通时也是不存在欧拉回路。
答案就是
$$f[s]=\frac{1}{g[s]}\ast \sum _{x\in s}f[x]g[sxorx]=\sum _{x\in s,y\in s,x|y=s,|x|+|y|=|s|}f[x]g[y]$$
显然这个是一个两个卷积的嵌套，我们设$F_i$表示一个多项式，其中$f(s)=F_{|s|}(s)$，$G_i$同理。
我们可以让
$$F_k=\sum _{i+j=k}{F}_i\ast G_j$$
此时就满足了$|x|+|y|=|s|$。里面的乘指或卷积，使用fwt解决，外面可以直接$O(n^2)$暴力卷积。
对于$F_k(s)$这个值，如果$|s|̸=k$，那么这个状态是不合法的。我们先进行逆运算，然后把不合法位置变为0，然后再变回来即可。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#define LL long long

const int maxn=23;
const int maxs=2100007;
const LL mod=998244353;

using namespace std;

int n,m,p,x,y;
int map[maxn][maxn],ru[maxn],bit[maxn],fa[maxn],numbit[maxs];
int f[maxn][maxs],g[maxn][maxs],sum[maxs],w[maxn];

int ksm(int x,int y)
{
    if (y==0) return 1;
    int c=ksm(x,y/2);
    c=((LL)c*(LL)c)%mod;
    if (y&1) c=((LL)c*(LL)x)%mod;
    return c;
}

int find(int x)
{
    if (!fa[x]) return x;
    return fa[x]=find(fa[x]);
}

void uni(int x,int y)
{
    int u=find(x),v=find(y);
    if (u==v) return;
    fa[u]=v;
}

int add(int x,int y)
{
	x=x+y;
	if (x>=mod) x-=mod;
	return x;
}

void fwt(int *a,int l,int r)
{
    if (l==r) return;
    int len=(r-l+1)/2,mid=l+len;
    fwt(a,l,mid-1);
    fwt(a,mid,r);
    for (int i=l;i<mid;i++) a[i+len]=add(a[i+len],a[i]);
}

void dwt(int *a,int l,int r)
{
    if (l==r) return;
    int len=(r-l+1)/2,mid=l+len;
    dwt(a,l,mid-1);
    dwt(a,mid,r);
    for (int i=l;i<mid;i++) a[i+len]=add(a[i+len],mod-a[i]);
}

int main()
{
    scanf("%d%d%d",&n,&m,&p);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        map[x][y]=1;
        map[y][x]=1;
    }
    for (int i=1;i<=n;i++) scanf("%d",&w[i]);
    bit[0]=1;
    for (int i=1;i<=n;i++) bit[i]=bit[i-1]*2;	
    for (int s=1;s<bit[n];s++)
    {
        for (int i=1;i<=n;i++)
        {
            if (s&bit[i-1])
            {
                sum[s]=add(sum[s],w[i]);
                numbit[s]++;
            }
            ru[i]=0,fa[i]=0;
        }		
        sum[s]=ksm(sum[s],p);
        for (int i=1;i<=n;i++)
        {
            if ((s&bit[i-1])==0) continue;
            for (int j=i+1;j<=n;j++)
            {
                if ((s&bit[j-1])==0) continue;
                if (map[i][j])
                {
                    ru[i]++,ru[j]++;
                    uni(i,j);
                }
            }
        }
        int F=0;
        for (int i=1;i<=n;i++)
        {
            if (s&bit[i-1])
            {
                if ((find(i)!=F) && (F))
                {
                    g[numbit[s]][s]=sum[s];
                    break;
                }
                else F=find(i);
                if (ru[i]&1)
                {
                    g[numbit[s]][s]=sum[s];
                    break;
                }
            }
        }
        sum[s]=ksm(sum[s],mod-2);
    }	
    for (int i=1;i<=n;i++) fwt(g[i],0,bit[n]-1);
    f[0][0]=1;
    for (int i=1;i<=n;i++)
    {
        fwt(f[i-1],0,bit[n]-1);
        for (int j=0;j<i;j++)
        {
            for (int s=0;s<bit[n];s++) f[i][s]=add(f[i][s],(LL)f[j][s]*(LL)g[i-j][s]%mod);
        }
        dwt(f[i],0,bit[n]-1);
        for (int s=0;s<bit[n];s++)
        {
            if (numbit[s]!=i) f[i][s]=0;
                         else f[i][s]=(LL)f[i][s]*(LL)sum[s]%mod;
        }
    }
    printf("%d\n",f[n][bit[n]-1]);
}