给出 n 个互不相同的正整数。
问存在多少个子集,使得子集中所有数的异或和是质数。
由于答案可能很大,请你输出对 109+7 取模后的结果。
输入格式
第一行包含整数 n。
第二行包含 n 个正整数。
输出格式
输出一个整数,表示满足条件的子集数量对 109+7 取模后的结果。
数据范围
1≤n≤5000,
1≤ 给定正整数 ≤5000。
输入样例:
3
1 2 3
输出样例:
4
思路分析
f[i][j]表示:所有只考虑前i个数,且异或和是j的方案数。
初始值:f[0][0]=1,f[0][j]=0(1<=j<=M)。
M=8192的原因:
5000=4096+x;(即能用13位数表示5000以内的所有数)。
异或规律:相同为0,不同为1。
异或和:所有的数进行异或运算。
所以能异或出来最大的就是13位全是1,即213-1=8191
代码
#include <iostream>
using namespace std;
const int M=8192,MOD=1e9+7;
int f[2][M+5];
bool isPrime(int x){
for(int i=2;i*i<=x;i++){//注意这里是有等于号的
if(x%i==0)//能整除某个数说明不是质数
return false;
}
return true;//是质数
}
int main()
{
int n,v;
cin>>n;
f[0][0]=1;//前0个数的质数和为0,所以有一种方案
for(int i=1;i<=n;i++){//注意点,这里不能从0开始
//因为从0开始,当i=0时,i&1=0,f[0]这一层是初始值
cin>>v;//输入第i个数
//这里这么写for(int j=M;j>=0;j--)也可以,但是要注意f数组不能开成f[2][M],会数组下标越界
for(int j=0;j<M;j++){//循环所有能构成的异或和情况
//i&1换成i%2也可以,但是i&1只需判断最后一位,效率更高
//其目的就是为了使用滚动数组,即奇数对应下标1,偶数对应下标0
//不使用滚动数组会超内存,因为5000*10000(8192的近似值)*4(字节)约等于200M
f[i&1][j]=f[(i-1)&1][j];//先初始化为不选第i个数
//异或是加减法的组合,异或一个数等于加上该数,再次异或等于减去该数
//因为是想异或v后等于j,所以异或v之前的异或和为j^v
if((j^v)<M){//特别注意点:<的 优先级大于^,所以能加括号就加括号,这里找了半小时
//当(j^v)<=M,有选i和不选i两种情况
f[i&1][j]=(f[(i-1)&1][j]+f[(i-1)&1][j^v])%MOD;
}
}
}
int res=0;
for(int i=2;i<M;i++){//让M近似10000,这样求质数也不是时间瓶颈,才10^6
if(isPrime(i)){//所以不需要用筛法求质数,筛法求质数还需要额外的空间,导致空间复杂度增加
res+=f[n&1][i];//统计所有异或和是质数的子集数
res%=MOD;//对1e9+7求余
}
}
cout<<res<<endl;//输出结果
return 0;
}