[FWT][DP] UOJ #310. 【UNR #2】黎明前的巧克力

$Solution$

DP很好想。
设$i$个巧克力的集合幂级数$f_i$，其中$f_{i,0}=1,f_{i,a_i}=2$。
答案就是这一堆东西对称差卷积起来。
暴力的做法就是把每个都FWT一下再乘起来。
集合幂级数$f$的沃尔什变换

$$\hat{f_S}=\sum_{T\subseteq U}(-1)^{|S\cap T|}f_T$$ 所以$\hat{f}_{i,S}$当$|S\cap a_i|$为偶数时为$3$，否则为$-1$。
考虑一个新的集合幂级数$g$，它是$f_i$的和。
那么$\hat{g}_S$即是一些$f_{*,S}$贡献了$-1$，一些贡献了$3$。
可以解方程解出$3$有$x={\hat{g}_{S}+n\over4}$个，那这里所有$\hat{f}_{*,S}$的乘积就是$3^x(-1)^{n-x}$。
重新FWT回去就是答案啦。
因为不允许有空集，要$-1$s。。。。

#include <bits/stdc++.h>
#define show(x) cerr << #x << " = " << x << endl
using namespace std;
typedef long long ll;
typedef pair<int, int> Pairs;

const int MOD = 998244353;
const int INV2 = (MOD + 1) / 2;
const int N = 1 << 21;

inline char get(void) {
    static char buf[100000], *S = buf, *T = buf;
    if (S == T) {
        T = (S = buf) + fread(buf, 1, 100000, stdin);
        if (S == T) return EOF;
    }
    return *S++;
}
template<typename T>
inline void read(T &x) {
    static char c; x = 0; int sgn = 0;
    for (c = get(); c < '0' || c > '9'; c = get()) if (c == '-') sgn = 1;
    for (; c >= '0' && c <= '9'; c = get()) x = x * 10 + c - '0';
    if (sgn) x = -x;
}

int n, m, mx, x;
int f[N];

inline void fwt(int* a, int f, int n) {
    static int x, y;
    for (int i = 1; i < n; i <<= 1)
        for (int j = 0; j < n; j += (i << 1))
            for (int k = 0; k < i; k++) {
                x = a[j + k]; y = a[j + k + i];
                a[j + k] = x + y;
                a[j + k + i] = x - y;
                if (f == -1) {
                    a[j + k] = (ll)a[j + k] * INV2 % MOD;
                    a[j + k + i] = (ll)a[j + k + i] * INV2 % MOD;
                }
            }
}
inline int pow(int a, int b) {
    int c = 1;
    while (b) {
        if (b & 1) c = (ll)c * a % MOD;
        b >>= 1; a = (ll)a * a % MOD;
    }
    return c;
}

int main(void) {
    freopen("1.in", "r", stdin);
    freopen("1.out", "w", stdout);
    read(n);
    for (int i = 1; i <= n; i++) {
        read(x); mx = max(mx, x);
        f[x] += 2; f[0] += 1;
    }
    for (m = 1; m <= mx; m <<= 1);
    fwt(f, 1, m);
    for (int i = 0; i < m; i++) {
        int x = (f[i] + n) / 4;
        f[i] = (ll)pow(3, x) * pow(MOD - 1, n - x) % MOD;
    }
    fwt(f, -1, m);
    cout << (f[0] - 1 + MOD) % MOD << endl;
    return 0;
}