题目描述
已知两个整数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);
}