HIHO #1044 : 状态压缩·一（状态压缩dp）

题目链接

使用dp[i][p1,p2…pm]：表示1到i位置的中，选择了i-m到i的一些p1,p2…pm位置，能获得的最大的垃圾。然后选择的位置用二进制 表示，状态转移的时候的，判断一下即可

getState函数也可以用二进制表示，j&((1< < m)-1)表示去掉j的二进制的第m位。

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<cstdlib>
#include<vector>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define LL long long
#define pb push_back
#define gcd __gcd

#define For(i,j,k) for(int i=(j);i<k;i++)
#define lowbit(i) (i&(-i))
#define _(x) printf("%d\n",x)

const int maxn = 1e3+10;
const int inf  = 1 << 28;

int w[maxn];
int dp[maxn][1<<11];

int getsum(int s){
    int cnt = 0;
    while(s){
        cnt+=(s&1);
        s>>=1;
    }
    return cnt;
}
int n,m,q;
/*
计算上一次的状态的最左边的一个
比如上一次[A...B],这一次就是[A+1,B+1],
等价的二进制表示 j&((1<<m)-1)
就是去掉一个数二进制的第m位 
*/
int getSate(int s){
    int ans = 0;
    for(int i=0;i<m-1;i++){
        ans|=s&(1<<i);
    }
    return ans;
}
int main(){

    while(~scanf("%d%d%d",&n,&m,&q)){
        for(int i=1;i<=n;i++){
            scanf("%d",&w[i]);
        }
        cl(dp,0);
        for(int i=1;i<=n;i++){
            for(int j=0;j<(1<<m);j++){
                if(getsum(j)>=q+1)continue;
                int t = getSate(j);
                if(getsum(t<<1|1)<=q){
                    dp[i][t<<1|1]=max(dp[i][t<<1|1],dp[i-1][j]+w[i]);
                }
                dp[i][t<<1]=max(dp[i][t<<1],dp[i-1][j]);
            }
        }
/*
        for(int i=1;i<=n;i++){
            for(int j=0;j<(1<<m);j++){
                printf(" %d", dp[i][j]);
            }
            printf("\n");
        }
*/
        int ans = 0;
        for(int i=0;i<(1<<m);i++)ans=max(ans,dp[n][i]);
        printf("%d\n",ans);

    }
    return 0;
}