1454. 异或和是质数的子集数（背包问题求方案数）

给出 n 个互不相同的正整数。

问存在多少个子集，使得子集中所有数的异或和是质数。

由于答案可能很大，请你输出对 10⁹+7 取模后的结果。

输入格式
第一行包含整数 n。

第二行包含 n 个正整数。

输出格式
输出一个整数，表示满足条件的子集数量对 10⁹+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，即2¹³-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;
}