[UOJ NOI Round #2 Day1]黎明前的巧克力

题目大意

给出n(n<=10^6)个在[0,10^6]内的整数a[1..n]，现要求从中选出两个不相交的集合，使得这两个集合的各自的异或和相等，可以存在有的数没有被选择，集合可以为空，但是两个集合不可以同时为空。
问选出两个这样的集合的方案数。
部分分：n和a[]都在1e3范围内

分析

一道喜闻乐见的fwt机智题。
先考虑部分分怎么做。发现xor相等等价于两个集合xor和等于0。那么设f[i][s]表示考虑了前i个点，xor和为s的方案数。转移很简单，要么不选第i+1个，要么把它加进其中一个集合，对s的影响是一样的，转移到$f[i+1][s\ xor\ a_{i+1}]$。
考虑使用fwt？很套路的做法是令每个元素为多项式，那么a[i]为$A_i(x)=1+2x^{a_i}$，然后每次加入一个，或者按权值分治做异或卷积。
但明显你把每个元素先transfer一遍，全部乘起来再untransfer，是O(n^2logn)的。
观察性质。
发现一个A(x)进行transfer之后，系数只有-1和3两种取值。其中3=1+2，-1=1-2，就是考虑x^0肯定对每一位都贡献1，而x^a[i]可能贡献±2。
考虑使用性质$tf(A+B)=tf(A)+tf(B)$。发现可以先把n个多项式累加起来，transfer一次，就能知道每个A变换后加起来的和。
这样的好处是可以解一元一次方程，找出每个位置-1和3的个数。
然后你就可以知道每个位置的累乘是多少了。
然后untransfer一下，就做完了。
时间复杂度O(nlogn)

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef double db;
#define fo(i,j,k) for(i=j;i<=k;i++)
#define fd(i,j,k) for(i=j;i>=k;i--)
#define cmax(a,b) (a=(a>b)?a:b)
#define cmin(a,b) (a=(a<b)?a:b)
const int N=2e6+5,M=1e6+5,mo=998244353,rt=3;
int m,i,n,rev4,rev2,cnt3,prod1,v,xs,half;
int a[N],pos[N],t[N];
int ksm(int x,int y)
{
    int ret=1;
    while (y)
    {
        if (y&1) ret=1ll*ret*x%mo;
        y>>=1;
        x=1ll*x*x%mo;
    }
    return ret;
}
void fwt(int *a,int l,int r,int s)
{
    if (l==r) return ;
    int m=l+r>>1;
    fwt(a,l,m,s);
    fwt(a,m+1,r,s);
    if (s==1) xs=1;else xs=rev2;
    half=r-l+1>>1;
    fo(i,l,m)
    {
        t[i]=1ll*(a[i]+a[i+half])*xs%mo;
        t[i+half]=1ll*(a[i]-a[i+half])*xs%mo;
    }
    fo(i,l,r) a[i]=t[i];
}
int main()
{
    freopen("t4.in","r",stdin);
//  freopen("t4.out","w",stdout);
    scanf("%d",&m);
    fo(i,1,m)
    {
        scanf("%d",pos+i);
        a[pos[i]]+=2;
        a[0]++;
    }
    n=1<<20;
    fwt(a,0,n-1,1);
    rev4=ksm(4,mo-2);
    rev2=ksm(2,mo-2);
    fo(i,0,n-1)
    {
        v=(a[i]+m)%mo;
        cnt3=1ll*v*rev4%mo;
        prod1=((m-cnt3)%2)?-1:1;
        a[i]=ksm(3,cnt3)*prod1;
    }
    fwt(a,0,n-1,0);
    printf("%d",(a[0]-1+mo)%mo);
}