[51nod1824]染色游戏

Description

有 n 个红球， m 个蓝球，从中取出 x 个红球和 y 个蓝球排成一排的得分是 rx⋅by ，其中 r0=b0=1 。
定义 f(t) 表示恰好取出 t 个球排成一排的所有可能局面的得分之和。
两个局面相同，当且仅当这两排球的个数相等，且在对应列位置上的颜色都是相同的。
求有多少 t (1≤t≤n+m) 使得 f(t) 是奇数，为了防止输出过大输出所有的t^2之和
n,m<=2^20

Solution

鞋垫手把手教我做题(人)
显然$f(t)=\sum_{i=0}^{t}r(i)b(t-i)C_t^i$
由于一些奇怪的定理我们知道$C_{n+m}^n$为奇数的充要条件是n&m=0
考虑规避组合数，我们设Ft,i表示二进制下和t差i个1的r之和，Gt,i表示b之和
F和G的求法可以参考上一场的F题“空间统计学”
那么我们可以先写出一个naive的式子

$$Ans(t)=\sum_{i=0}^{|t|}F[t][i]*G[t][|t|-i]$$
然而这样显然会算重，我们需要考虑如何减重
出现重复就是F所记录的某一个状态和G所出现的某一个状态的and值不为0
那么我们考虑把出现的位置给减掉，那么 $$Ans(t)=\sum_{T\subseteq t}(-1)^{|T|}\sum_{i=|T|}^{|t|-|T|}F[t-T][i-|T|]*G[t-T][|t|-i-|T|]$$
但是直接子集枚举的复杂度是无法承受的，我们需要另想其他办法
先把后面写成卷积 $$\sum_{i=0}^{|t|-2*|T|}F[t-T][i]*G[t-T][|t|-2*|T|-i]$$
考虑维护 $$H[t][i]=\sum_{j=0}^{i}F[t][j]*G[t][i-j]$$
那么 $$Ans(t)=\sum_{T\subseteq t}(-1)^{|T|}H[t-T][|t|-2*|T|]$$
后面这个东西可以用分治求出来，分治最高位，把最高位是0的合并到最高位是1的。
那么这道题就做完了，总复杂度O(n log^2 n)
注意这道题卡空间，注意我们并不需要真的求出Ans，只需要求出Ans的奇偶性，那么我们可以只开bool类型来存储，所有的加法用^，所有的乘法用&
还有我们发现n+m的最高位可以有21位，直接开2^21的数组会爆空间
但是我们知道只有所有满足存在一个i使得(t-i)&i=0的位置是有用的
发现这样的t最大就是2^20-1
所以我们直接处理2^20-1以内的数据就好了
空间复杂度O(n log n)

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

const int N=(1<<20)+5;

int n,m,lg1,lg2,lg,all,two[21];
bool f[N][21],g[N][21],h[N][21];

void init() {
    scanf("%d%d",&n,&m);

    char ch;for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
    f[0][0]=1;
    fo(i,1,n) {
        f[i][0]^=(ch-'0')&1;
        ch=getchar();
    }
    int len=1;lg1=0;
    for(;len<=n;len<<=1) lg1++;

    for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
    g[0][0]=1;
    fo(i,1,m) {
        g[i][0]^=(ch-'0')&1;
        ch=getchar();
    }

    len=1;lg2=0;
    for(;len<=m;len<<=1) lg2++;

    lg=max(lg1,lg2);
    two[0]=1;fo(i,1,lg) two[i]=two[i-1]<<1;
    all=two[lg]-1;
}

inline void calc(int l,int r) {
    if (l==r) return;
    int mid=l+r>>1;
    calc(l,mid);calc(mid+1,r);
    fo(i,mid+1,r) 
        fo(j,2,lg)
            h[i][j]^=h[i-mid+l-1][j-2];
}

void solve() {
    ll Ans=0;
    fo(t,1,min(all,n+m)) {
        int cnt=0;
        fo(j,0,lg-1) if (t&two[j]) cnt++; 
        Ans+=(ll)h[t][cnt]*t*t;
    }
    printf("%lld\n",Ans);
}

int main() {
    init();
    fo(j,1,lg)
        fd(s,all,0) 
            if (!(s&two[j-1]))
                fo(i,0,j-1) f[s|two[j-1]][i+1]^=f[s][i];
    fo(j,1,lg)
        fd(s,all,0) 
            if (!(s&two[j-1]))
                fo(i,0,j-1) g[s|two[j-1]][i+1]^=g[s][i];
    fo(i,0,all)
        fo(j,0,lg) 
            fo(k,0,j)
                h[i][j]^=f[i][k]&g[i][j-k];
    calc(0,all);solve();
    return 0;
}