[51nod1301]集合异或和

题目描述

已知两个整数N与M,你需要构造两个整数集合X与Y，且需要满足以下要求：
（1）对所有的xi∈X，满足1<=xi<=N;对所有的yj∈Y，满足1<=yj<=M; （X与Y可以为空集）
（2）X∩Y=Φ;（但不要求集合X与Y的元素个数，只要两者没有交集即可）

不妨设构造后的集合X含有n个元素，而集合Y有m个元素，令 A=x1 xor x2 xor x3 xor … xor xn， B=y1 xor y2 xor y3 … xor ym。其中，xor 为异或运算，并且 称A与B分别是集合X与集合Y的异或和。如果一个集合为空集，那它的异或和被认为是0。

求对于给定的N与M符合要求（1）、（2）且满足$A<B$的集合X、Y的不同构造方法个数，输出结果 mod 10^9 + 7。其中，两种构造方案被认为是不同的，当且仅当两种方案中的集合X或集合Y包含的元素集不完全相同。

例如：N=2，M=2，则符合条件的X与Y有如下四组：
X={} and Y={1}
X={} and Y={2}
X={} and Y={1, 2}
X={1} and Y={2}

DP

枚举A xor B的第一个非0位x。
那么显然B的第x位为1，A的第x位为0，就可以满足$A<B$
设f[i,j,0~1]表示分配完了前i个数，目前异或值为j，B的第x位是0或1的方案数。
转移就是看第i+1个数给A还是B。
统计答案的时候，显然为了满足异或和最高位为x，我们要找所有在[2^x,2^(x+1)-1]范围的i，加上f[max(n,m),i,1]（B的第x位必须为1）
复杂度为2次方带log，有点卡。
卡常数就用模优化，只有当值>=模数是再用减法去减，详见代码。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=2000+10,mo=1000000007;
int f[2][maxn*2][2];
int i,j,k,l,t,n,m,mx,ans,x,zjy,now;
int main(){
    scanf("%d%d",&n,&m);
    mx=max(n,m);
    zjy=floor(log(mx)/log(2));
    fo(x,0,zjy){
        now=0;
        fo(j,0,2048)
            f[now][j][0]=f[now][j][1]=0;
        f[now][0][0]=1;
        fo(i,1,mx){
            now=1-now;
            fo(j,0,2048)
                f[now][j][0]=f[now][j][1]=0;
            if ((i&(1<<x))>0) t=1;else t=0;
            fo(j,0,2048){
                l=j^i;
                if (i<=n){
                    f[now][l][0]=(f[now][l][0]+f[1-now][j][0]);
                    f[now][l][1]=(f[now][l][1]+f[1-now][j][1]);
                    if (f[now][l][0]>=mo) f[now][l][0]-=mo;
                    if (f[now][l][1]>=mo) f[now][l][1]-=mo;
                }
                if (i<=m){
                    f[now][l][t]=(f[now][l][t]+f[1-now][j][0]);
                    f[now][l][t^1]=(f[now][l][t^1]+f[1-now][j][1]);
                    if (f[now][l][t]>=mo) f[now][l][t]-=mo;
                    if (f[now][l][t^1]>=mo) f[now][l][t^1]-=mo;
                }
                f[now][j][0]=(f[now][j][0]+f[1-now][j][0]);
                f[now][j][1]=(f[now][j][1]+f[1-now][j][1]);
                if (f[now][j][0]>=mo) f[now][j][0]-=mo;
                if (f[now][j][1]>=mo) f[now][j][1]-=mo;
            }
        }
        fo(i,(1<<x),(1<<x)*2-1) ans=(ans+f[now][i][1])%mo;
    }
    printf("%d\n",ans);
}