状态压缩·一

题目传送：#1044 : 状态压缩·一

AC代码：

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <cctype>
#include <cstdio>
#include <string>
#include <vector>
#include <complex>
#include <cstdlib>
#include <cstring>
#include <fstream>
#include <sstream>
#include <utility>
#include <iostream>
#include <algorithm>
#include <functional>
#define LL long long
#define INF 0x7fffffff
using namespace std;

int n, m, q;
int w[1005];

int dp[1005][1030];//dp[i][j]表示选到第i个位置时j状态能够取得的最大值
int cnt[1030];//代表每个数的位数上的1的个数

int main() {
    cnt[0] = 0, cnt[1] = 1;
    for(int i = 2; i < 1030; i ++) cnt[i] = cnt[i >> 1] + cnt[i & 1];

    scanf("%d %d %d", &n, &m, &q);
    for(int i = 1; i <= n; i ++) {
        scanf("%d", &w[i]);
    }

    int ans = 0;
    int d = 1 << m;
    for(int i = 1; i <= n; i ++) {
        for(int j = 0; j < (1 << m); j ++) {
            if(cnt[j] <= q) dp[i][j] = max(dp[i-1][j >> 1], dp[i-1][(j >> 1) + (1 << (m - 1))]) + (j & 1) * w[i];
            ans = max(ans, dp[i][j]);
        }
    }

    printf("%d\n", ans);
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。