hihoCoder 1044 状态压缩·一

题目链接：http://hihocoder.com/problemset/problem/1044?sid=784987

题意：有n个数，可以选取一些数，但是在每个连续区间长度为m的区间中的被选中数的个数不能超过q个，问可以选取到的和的最大值是多少。

思路：我们用dp[i][j]表示在第i个位置上，j是一个二进制状态，表示包括当前位置的前M-1个位置上的数是否被选取。
那么dp[i][s] = max( dp[i-1][last]+w[i] , dp[i][s] ),我们枚举可行的last,并判断last中含有1的个数，如果last中含有q个1，第i个位置只能不选了；否则既可以不选也可以选。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <utility>
using namespace std;

#define rep(i,j,k) for (int i=j;i<=k;i++)
#define Rrep(i,j,k) for (int i=j;i>=k;i--)

#define Clean(x,y) memset(x,y,sizeof(x))
#define LL long long
#define ULL unsigned long long
#define inf 0x7fffffff
#define mod %100000007

int n,m,q;

int dp[1009][1<<9]; //在第i个位置上 当前向后的m-1个位置的状态
int w[1009];
int num[1<<9];
int state[1<<9];
int S;
void init()
{
    Clean(num,0);
    S = 0;
    int uplim = 1 << (m-1);
    rep(i,0,uplim-1)
    {
        int ans = 0;
        int temp = i;
        while( temp )
        {
            temp-=temp&(-temp);
            ans++;
        }
        num[i] = ans;
        if ( ans <= q ) state[++S] = i;//预处理出可行状态
    }
}


int main()
{
    scanf("%d%d%d",&n,&m,&q);
    rep(i,1,n) scanf("%d",&w[i]);
    if ( m == q )
    {
        int ans = 0;
        rep(i,1,n) ans += w[i];
        printf("%d\n",ans);
        return 0;
    }
    init();
    int add = 1<<(m-2);
    Clean(dp,0);
    dp[1][1<<(m-2)] = w[1];
    rep(i,2,n)
    {
        rep(j,1,S) //枚举i-1的状态
        {
            int nexts = state[j]>>1;
            if ( num[ state[j] ] < q )
                dp[i][nexts+add] = max( dp[i][nexts+add] , dp[i-1][state[j]] + w[i] );
            dp[i][nexts] = max( dp[i][nexts] , dp[i-1][state[j]]);

        }
    }
    int ans = 0;
    rep(i,1,S) ans = max( ans , dp[n][ state[i] ] );
    printf("%d\n",ans);
    return 0;
}