hdu 4901 The Romantic Hero（计数dp）

来源：互联网发布：淘宝换货要申请吗编辑：程序博客网时间：2024/05/01 13:41

http://acm.hdu.edu.cn/showproblem.php?pid=4901

dp1[i][j] : 0~i中的数进行异或得到j的种类数，不一定包含i。

dp2[i][j]：i~n-1中的数进行与得到j的种类数，不一定包含i。

dp3[i][j] : i~n-1中的数进行与得到j的种类数，包含i，防止计数重复。

注意中间取余

#include <stdio.h>#include <iostream>#include <map>#include <set>#include <stack>#include <vector>#include <math.h>#include <string.h>#include <queue>#include <string>#include <stdlib.h>#include <algorithm>#define LL long long#define _LL __int64#define eps 1e-12#define PI acos(-1.0)#define C 240#define S 20using namespace std;const int maxn = 1010;const int maxm = 1030;const int mod = 1000000007;int dp1[maxn][maxm],dp2[maxn][maxm],dp3[maxn][maxm];int a[maxn];int main(){int test;scanf("%d",&test);while(test--){int n;scanf("%d",&n);for(int i = 0; i < n; i++)scanf("%d",&a[i]);memset(dp1,0,sizeof(dp1));memset(dp2,0,sizeof(dp2));memset(dp3,0,sizeof(dp3));dp1[0][a[0]]++;for(int i = 1; i < n-1; i++){dp1[i][a[i]]++;for(int j = 0; j < 1025; j++){if(dp1[i-1][j]){dp1[i][j] += dp1[i-1][j];dp1[i][j] %= mod;dp1[i][j^a[i]] += dp1[i-1][j];dp1[i][j^a[i]] %= mod;}}}dp2[n-1][a[n-1]] = 1;dp3[n-1][a[n-1]] = 1;for(int i = n-2; i >= 0; i--){dp2[i][a[i]]++;dp3[i][a[i]]++;for(int j = 0; j < 1025; j++){if(dp2[i+1][j]){dp2[i][j] += dp2[i+1][j];dp2[i][j] %= mod;dp2[i][j&a[i]] += dp2[i+1][j];dp2[i][j&a[i]] %= mod;dp3[i][j&a[i]] += dp2[i+1][j];dp3[i][j&a[i]] %= mod;}}}int ans = 0;for(int i = 0; i < n-1; i++){for(int j = 0; j < 1025; j++){if(dp1[i][j] && dp3[i+1][j]){ans += (((LL)dp1[i][j]*dp3[i+1][j])%mod);ans %= mod;}}}printf("%d\n",ans);}return 0;}

0 0