ZOJ 3164 Cookie Choice（终极背包）

题意：

给出N类物品，有价值，数量，价格

还有一些限制组，限制的组里面最多只能选一类物品，问D钱最多能获得的价值，并且D钱必须花完

思路：

看完背包9讲，感觉这一题还是不难理解的。首先，对于没有组的限制的X类物品，进行一遍背包（01背包、完全背包、多重背包）。

其次，对于有组的限制的物品，就需要再次考虑下了。这次要用到泛化背包和分组背包的思想。每一组只能选择组里面的一类物品。

于是试图考虑，如果我分给这个组V(range from 0 to D)的容量，能够最多拿到多少价值。因为这个组有X类物品，只能选择一类，

所以需要先对这个组的每一类物品进行一次01背包，获得分配给这类物品V的容量取得的最大价值，然后再对这个组的X类物品进行分组背包。

最后，针对上述组进行一次分组背包，问题得以解决。

（第一次开始自己尝试描述算法思路，Orz Orz，坚持坚持）

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <algorithm>



using namespace std;



#define max(a,b) (((a) > (b)) ? (a) : (b))



const int MAXN = 1500;

const int INF = 1e9;

int Ki[MAXN], Ei[MAXN], Pi[MAXN];

int w[10][MAXN], f[MAXN];

int n, d, g;



void init_input()

{

    for (int i = 1; i <= n; ++i)

        scanf("%d %d %d", &Ki[i], &Ei[i], &Pi[i]);



    scanf("%d%*c", &g);

    memset(f, 0, sizeof(f));

    char s[1000];

    for (int i = 1; i <= g; ++i)

    {

        gets(s);

        int j = 0;

        while (j < strlen(s))

        {

            if ('1' <= s[j] && s[j] <= '9')

            {

                int sum = 0;

                while ('0' <= s[j] && s[j] <= '9')

                    sum = 10 * sum + s[j] - '0', ++j;

                f[sum] = i;

            } 

            else 

                ++j;

        }

    }

}



void init_dp(int dp[])

{

    fill(dp, dp + d + 1, -INF);

    dp[0] = 0;

}



void complete_pack(int w, int v, int dp[])

{

    for (int i = w; i <= d; ++i)

        if (dp[i-w] > -INF)             // 必须装满的时候，是需要这么一个判断的。-INF表示没有合法解

            dp[i] = max(dp[i], dp[i-w] + v);

}



void zero_one_pack(int w, int v, int dp[])

{

    for (int i = d; i >= w; --i)

        if (dp[i-w] > -INF)

            dp[i] = max(dp[i], dp[i-w] + v);

}



void multiple_pack(int w, int v, int n, int dp[])

{

    int i = 1, c = n;



    while (i < c)

    {

        zero_one_pack(i * w, i * v, dp);

        c -= i;

        i *= 2;

    }

    if (c)

        zero_one_pack(c * w, c * v, dp);

}



int main()

{

    while (scanf("%d %d", &n, &d) != EOF)

    {

        init_input();



        int dp[MAXN], temp[MAXN];



        init_dp(dp);

        for (int i = 1; i <= g; ++i)

            init_dp(w[i]);



        for (int i = 1; i <= n; ++i)

        {

            if (f[i])

                init_dp(temp);



            if (!Ki[i] || Ki[i] * Pi[i] >= d)

                complete_pack(Pi[i], Ei[i], f[i] ? temp : dp);

            else

                multiple_pack(Pi[i], Ei[i], Ki[i], f[i] ? temp : dp);



            if (f[i])   

                for (int j = 0; j <= d; ++j)

                    w[f[i]][j] = max(w[f[i]][j], temp[j]);

        }

        for (int i = 1; i <= g; ++i)    // w[g][d] ： g个组，每个组有d个物品，每个组最多选择其中一个

            for (int j = d; j >= 0; --j)

                for (int k = 0; k <= j; ++k)

                    if (w[i][k] > -INF && dp[j-k] > -INF)

                        dp[j] = max(dp[j], dp[j-k] + w[i][k]);



        if (dp[d] >= 0)

            printf("%d\n", dp[d]);

        else 

            printf("i'm sorry...\n");

    }

    return 0;

}