hihocoder #1044 : 状态压缩·一

描述

小Hi和小Ho本着礼让他们的心情——当然还因为本来他们买的就是站票,老老实实的呆在两节车厢的结合处。他们本以为就能够这样安稳抵达目的地,但 事与愿违,他们这节车厢的乘务员是一个强迫症,每隔一小会总是要清扫一次卫生,而时值深夜,大家都早已入睡,这种行为总是会惊醒一些人。而一旦相邻的一些 乘客被惊醒了大多数的话,就会同乘务员吵起来,弄得大家都睡不好。

将这一切看在眼里的小Hi与小Ho决定利用他们的算法知识,来帮助这个有着强迫症的乘务员——在不与乘客吵起来的前提下尽可能多的清扫垃圾。

小Hi和小Ho所处的车厢可以被抽象成连成一列的N个位置,按顺序分别编号为1..N,每个位置上都有且仅有一名乘客在休息。同时每个位置上都有一 些垃圾需要被清理,其中第i个位置的垃圾数量为Wi。乘务员可以选择其中一些位置进行清理,但是值得注意的是,一旦有编号连续的M个位置中有超过Q个的位 置都在这一次清理中被选中的话(即这M个位置上的乘客有至少Q+1个被惊醒了),就会发生令人不愉快的口角。而小Hi和小Ho的任务是,计算选择哪些位置 进行清理,在不发生口角的情况下,清扫尽可能多的垃圾。

提示一:无论是什么动态规划,都需要一个状态转移方程!

提示二:好像什么不对劲?状态压缩哪里去了?

输入

每个测试点(输入文件)有且仅有一组测试数据。

每组测试数据的第一行为三个正整数N、M和Q,意义如前文所述。

每组测试数据的第二行为N个整数,分别为W1到WN,代表每一个位置上的垃圾数目。

对于100%的数据,满足N<=1000, 2<=M<=10,1<=Q<=M, Wi<=100

输出

对于每组测试数据,输出一个整数Ans,表示在不发生口角的情况下,乘务员最多可以清扫的垃圾数目。

样例输入
5 2 1
36 9 80 69 85 
样例输出
201

状压dp学习中,发现不好懂。找了篇结题报告,把其中dp数组的计算输出了。

5 2 1

36 9 80 69 85

如果dp[0][0] != -1
计算dp[1][0]  dp[1][1]

如果dp[0][1] != -1
计算dp[1][2]  dp[1][3]

如果dp[0][2] != -1
计算dp[1][0]  dp[1][1]

如果dp[0][3] != -1
计算dp[1][2]  dp[1][3]

如果dp[1][0] != -1
计算dp[2][0]  dp[2][1]

如果dp[1][1] != -1
计算dp[2][2]  dp[2][3]

如果dp[1][2] != -1
计算dp[2][0]  dp[2][1]

如果dp[2][0] != -1
计算dp[3][0]  dp[3][1]

如果dp[2][1] != -1
计算dp[3][2]  dp[3][3]

如果dp[2][2] != -1
计算dp[3][0]  dp[3][1]

如果dp[3][0] != -1
计算dp[4][0]  dp[4][1]

如果dp[3][1] != -1
计算dp[4][2]  dp[4][3]

如果dp[3][2] != -1
计算dp[4][0]  dp[4][1]

如果dp[4][0] != -1
计算dp[5][0]  dp[5][1]

如果dp[4][1] != -1
计算dp[5][2]  dp[5][3]

如果dp[4][2] != -1
计算dp[5][0]  dp[5][1]

dp数组:
0  0  0  0

0  36  0  -1

0  9  36  -1

36  116  9  -1

36  105  116  -1

116  201  105  -1

最后结果=201

#include <stdio.h>
#include <iostream>
#include <cstring>
using namespace std;

const int MAX_N = 1001;
const int MAX_M = 10;//最多连续10个位置

int N, M, Q;//不能超过Q
int w[MAX_N];//保存每个位置的垃圾
int dp[MAX_N][1<<MAX_M];//dp[][]大小 n*(2^m)
bool ok[1<<MAX_M];//???

void init()
{
    memset(ok, false, sizeof(ok));//ok标记数组初始化
    int val, cnt;
    for(int i=0; i<(1<<M); ++i){
        val=i; cnt=0;
        while(val>0){
            cnt+=(val&1);
            val>>=1;
        }
        ok[i]=(cnt<=Q);
    }
    /*
      这段预处理的作用:0到(2^m-1)的所有数可以转换成二进制
      转换成的二进制表示中1的个数时候大于Q,如果大于Q,则说明
      当前的这种状态是不可取的,可以忽略
    */
}

void solve()
{
    memset(dp, -1, sizeof(dp));//dp全部初始化为-1
    memset(dp[0], 0, sizeof(dp[0]));//第一行初始化为0

    for(int i=1; i<=N; ++i)
    {
        for(int j=0; j<(1<<M); ++j)
        if (dp[i-1][j] != -1)// !=-1的话 说明前面有过计算出一个优解
        {
            printf("如果dp[%d][%d] != -1\n", i-1, j);

            int s0 = ((j<<1) & ((1<<M) - 1));
            int s1 = ((j<<1 | 1) & ((1<<M) - 1));
            printf("计算dp[%d][%d]  dp[%d][%d]\n\n\n", i, s0, i, s1);

            dp[i][s0] = max(dp[i][s0], dp[i-1][j]);
            if (ok[s1]){
                dp[i][s1] = max(dp[i][s1], dp[i-1][j] + w[i]);
            }
        }
    }
    for(int i=0; i<=N; i++){
        for(int j=0; j<(1<<M); j++){
            printf("%d  ", dp[i][j]);
        }printf("\n\n");
    }
    int res=0;
    for (int i = 0; i < (1<<M); ++i){
        res = max(res, dp[N][i]);
    }
    cout << res << endl;
}

int main()
{
    while(cin>>N>>M>>Q)
    {
        init();
        for(int i=1; i<=N; ++i) {
            cin >> w[i];
        }//读入数据 每个位置的垃圾
        solve();
    }
    return 0;
}

 



posted @ 2015-08-28 10:37  我喜欢旅行  阅读(229)  评论(0编辑  收藏  举报