HDU 5119 Happy Matt Friends(dp)

Description
给出n个整数,问从中选取任意数量的数做异或和结果不小于m的方法数
Input
第一行为一整数T表示用例组数,每组用例第一行为两个整数n和m,第二行n个整数ai(1<=n<=40,0<=m<=10^6,0<=ai<=10^6)
Output
对于每组用例,输出从这n个数中选取任意数量的数做异或和结果不小于m的方法数
Sample Input
2
3 2
1 2 3
3 3
1 2 3
Sample Output
Case #1: 4
Case #2: 2
Solution
以dp[i][j]表示从前i个数中选取任意数量的数做异或和结果为j的方法数,那么我们有转移方程dp[i][j]=dp[i-1][j]+dp[i][j^a[i]],第二维比较大而第一维递推过程只由i-1推到i所以第一维用滚动数组节省空间,那么结果就是sum(dp[n][k]),其中m<=k< 1<<20(因为n个不超过10^6的数异或和最多为(1<<20)-1)
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn (1<<20)+5
typedef long long ll;
int T,n,m,a[44],dp[2][maxn];
int main()
{
    scanf("%d",&T);
    for(int Case=1;Case<=T;Case++)
    {
        scanf("%d%d",&n,&m);
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        dp[0][0]=1;
        ll ans=0;
        for(int i=1;i<=n;i++)
            for(int j=0;j<(1<<20);j++)
                dp[i%2][j]=dp[(i-1)%2][j]+dp[(i-1)%2][j^a[i]];
        for(int i=m;i<(1<<20);i++)ans+=dp[n%2][i];
        printf("Case #%d: %lld\n",Case,ans);
    }
    return 0;
}

你可能感兴趣的:(HDU 5119 Happy Matt Friends(dp))