HDOJ 3535 AreYouBusy （背包 每组至少一个）

题目链接：(—_—) zZ

题目大意：有n种工作的集合， 每种集合有一种类型， 0为最少在这个集合中选一个工作， 1为最多在这个集合中选一个工作， 2为随便选， 做每个工作要花一定的时间和得到一定的幸福度， 求在t时间内获得的最大幸福度

思路：类型为0时是分组背包的变形 ， 为1时就是分组背包， 为2是为0-1背包，分别根据背包类型求出就可

Ps：无限wa

code：

#include <stdio.h>
#include <string.h>
struct node
{
    int t, h;
}pack[5][100002];
int c = 0, n = 0, t = 0,dp[102][102], num[5][102], index1[5], ind[102];
void zeropack( int in, int start)//至少选一个
{
    int i = 0, j = 0, k  = 0, tpval = 0;
    for(i = 0; i<num[0][in]; i++)
    {
        for(j = t; j>=pack[0][start+i].t; j--)
        {
            if(dp[c][j-pack[0][start+i].t] != -1)
                dp[c][j] = dp[c][j]>dp[c][j-pack[0][start+i].t]+pack[0][start+i].h? dp[c][j]:dp[c][j-pack[0][start+i].t]+pack[0][start+i].h;
            if(dp[c-1][j-pack[0][start+i].t] != -1)
                dp[c][j] = dp[c][j]>dp[c-1][j-pack[0][start+i].t]+pack[0][start+i].h? dp[c][j]:dp[c-1][j-pack[0][start+i].t]+pack[0][start+i].h;
        }
    }
}

void onepack(int in, int start)//最多选一个 分组背包问题
{
    int i = 0, j = 0, k = 0, tpval = 0;
    for(i = 0; i<=t; i++)
        dp[c][i] = dp[c-1][i];
    for(i = t; i>=0; i--)
    {
        for(j = 0; j<num[1][in]; j++)
            if(pack[1][start+j].t<=i && dp[c-1][i-pack[1][start+j].t] != -1)
                dp[c][i] = dp[c][i]>dp[c-1][i-pack[1][start+j].t]+pack[1][start+j].h? dp[c][i]:dp[c-1][i-pack[1][start+j].t]+pack[1][start+j].h;
    }
}

void twopack(int in, int start)//0-1背包
{
    int i = 0, j =0, k = 0, tpval = 0;
    for(i = 0; i<=t; i++)
        dp[c][i] = dp[c-1][i];
    for(i = 0; i<num[2][in]; i++)
    {
        for(j = t; j>=pack[2][start+i].t; j--)
        {
            if(dp[c][j-pack[2][start+i].t] != -1 )
                dp[c][j] = dp[c][j]>dp[c][j-pack[2][start+i].t]+pack[2][start+i].h? dp[c][j]:dp[c][j-pack[2][start+i].t]+pack[2][start+i].h;
        }
    }
    c++;
}

int main()
{
    int i = 0, j = 0, m = 0, s = 0, start = 0, ans = -1;
    while(scanf("%d %d", &n, &t) != EOF)
    {
        c = 1;
        memset(index1, 0, sizeof(index1));
        memset(ind, 0, sizeof(ind));
        memset(dp, -1, sizeof(dp));
        for(i = 0; i<=t; i++)
            dp[0][i] = 0;
        for(i = 0; i<n; i++)
        {
            scanf("%d %d", &m, &s);
            num[s][index1[s]++] = m;//index1[s]表示有多少类为s的组
            for(j =0; j<m; j++, ind[s]++)
            {
                scanf("%d", &pack[s][ind[s]].t);
                scanf("%d", &pack[s][ind[s]].h);
            }
        }
            start = 0;
         for(j = 0; j<index1[0]; j++)//一共有多少组类属于0的
          {
          zeropack(j, start);
              start += num[0][j];
              c++;
          }
          start = 0;
          for(j = 0; j<index1[1]; j++)
          {
              onepack(j, start);
              start += num[1][j];
              c++;
          }
          start = 0;
          for(j = 0; j<index1[2]; j++)
          {
              twopack(j, start);
              start += num[2][j];
          }
         printf("%d\n", dp[c-1][t]);
  }
    return 0;
}