题目: AcWing 1454. 异或和是质数的子集数
给出 n 个互不相同的正整数。
问存在多少个子集,使得子集中所有数的异或和是质数。
由于答案可能很大,请你输出对 109+7 取模后的结果。
输入格式
第一行包含整数 n。
第二行包含 n 个正整数。
输出格式
输出一个整数,表示满足条件的子集数量对 109+7 取模后的结果。
数据范围
1≤n≤5000,
1≤ 给定正整数 ≤5000。
输入样例:
3
1 2 3
输出样例:
4
题目分析:
首先因为范围是5000,所以我们可以知道最大的异或和就是213-1=8192(就是每位二进制取1)
范围不大我们可以使用背包问题,每个数字有两种情况选或者不选。
状态表示:f ( i, j)表示前i个数字,异或和是j的方案数
状态转移:
1.如果不选,那么 f ( i, j) = f ( i - 1, j) 很容易想到就是上一次的情况
2.如果选,那么f ( i , j) = f( i - 1, j ^ a[ i]) 很容易想到就是上一次异或了a[i]的情况(因为 j ^ a[ i] ^ a[ i] = j,所以找到上一次异或的情况,再异或就是j)
注意:内存超限,需要用到滚动数组
因为我们不知道 j ^ a[ i ]是大于j还是小于j,不能优化为一维数组。
如果我们开f[ 5000][8192]二维数组则是5000*9000*4=1.8*108=100多MB
内存限制64MB,超出。
可以使用二维滚动数组,因为计算当前状态只需要上一个状态,我们可以用 i & 1来判断奇偶
#include <iostream>
using namespace std;
const int N = 5010, M = 8192 ,MOD = 1e9+7;
int a[N];
int f[2][M];
int n;
bool is_prime(int x) // 判断质数
{
for(int i=2;i*i<=x;i++)
if(x%i==0)return false;
return true;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i];
f[0][0]=1;
for(int i=1;i<=n;i++)
for(int j=0;j<M;j++)
{
f[i&1][j]=f[i-1&1][j];
if((j^a[i])<M)f[i&1][j]=(f[i&1][j]+f[i-1&1][j^a[i]])%MOD;
}
int res=0;
for(int i=2;i<M;i++)
if(is_prime(i))res=(res+f[n&1][i])%MOD;
cout<<res<<endl;
return 0;
}