HDOJ 4901 - The Romantic Hero 简单dp

最新推荐文章于 2016-01-13 09:04:38 发布

原创最新推荐文章于 2016-01-13 09:04:38 发布 · 777 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

动态规划专栏收录该内容

130 篇文章

订阅专栏

本文探讨了一列数中寻找满足异或与操作相等的子集S和T的数量，涉及复杂度优化和计数策略。

题意:

有一列n个数(n<=1000,每个数的范围是[0,1024]),现在要从这列数中取出两个集合S与T,其中S中每个元素要在T的左边,现在对S的所有数做异或操作,对T中所有数做与操作,若算出了相同的数,那么当前的S,T就是所需要的,问有多少组S,T符合要求,答案对10^9+7取摸。

题解:

关键是T中的与操作,所知两个数做与操作得到的数不会比这两个数大,所以可以限制了数据的范围,首先用dp[x][y]代表从x开始往后这些数中取一部分做与操作,能得到y的个数.这个过程的复杂度是O(n*1024).再从头开始做异或操作,注意的是两个数异或得到的数是可能比这两个数大的,若计算中算出超过1024可以直接舍弃.那么取值的时候为了防止重复统计,每次都是保证了前半部分会取第i位后能得到某个异或值x的个数乘上从i+1位(不一定取i+1位)往后能通过与操作得到值x的个数.

Program:

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#define MAXN 1050
#define ll long long
#define oo 1000000007
using namespace std;
ll a[MAXN],s[MAXN],p[MAXN],dp[MAXN][MAXN];
int main()
{
       ll cases,n,i,x,ans;
       freopen("input.txt","r",stdin);
       freopen("output.txt","w",stdout);
       scanf("%I64d",&cases);
       while (cases--)
       {
               scanf("%I64d",&n); 
               for (i=1;i<=n;i++) scanf("%I64d",&a[i]);
               memset(dp,0,sizeof(dp));
               for (i=n;i>=1;i--)
               {
                    memcpy(dp[i],dp[i+1],sizeof(dp[i])); 
                    for (x=0;x<=1024;x++)
                       dp[i][x&a[i]]=(dp[i][x&a[i]]+dp[i+1][x])%oo;
                    dp[i][a[i]]=(dp[i][a[i]]+1)%oo;
               }
               memset(p,0,sizeof(p));
               ans=0;
               for (i=1;i<=n;i++)
               {
                    memset(s,0,sizeof(s));
                    for (x=0;x<=1024;x++)
                       if ((x^a[i])<=1024)
                          s[x^a[i]]=(s[x^a[i]]+p[x])%oo;
                    s[a[i]]=(s[a[i]]+1)%oo;
                    for (x=0;x<=1024;x++)
                       ans=(ans+s[x]*dp[i+1][x])%oo;
                    for (x=0;x<=1024;x++) p[x]=(p[x]+s[x])%oo;                     
               }
               printf("%I64d\n",ans);
       }
       return 0;
}