洛谷 P2409 Y的积木

题目背景

Y是个大建筑师，他总能用最简单的积木拼出最有创意的造型。

题目描述

Y手上有n盒积木，每个积木有个重量。现在他想从每盒积木中拿一块积木，放在一起，这一堆积木的重量为每块积木的重量和。现在他想知道重量和最小的k种取法的重量分别是多少。（只要任意更换一块积木，就视为一种不同的取法。如果多种取法重量总和一样，我们需要输出多次。）

输入输出格式
输入格式：

第一行输入两个整数，n,k，意义如题目所描述。

每组数据接下来的n行，第一个整数为mi，表示第i盒积木的数量，在同一行有mi个整数，分别表示每个积木的重量。

输出格式：

一行，重量最小的k种取法的重量，要求对于每个数据，从小到大输出

输入输出样例

输入样例#1：
3 10
4 1 3 4 5
3 1 7 9
4 1 2 3 5

输出样例#1：
3 4 5 5 6 6 7 7 7 7

说明

对于30%的数据：2<=mi<=10,1<=n<=10

对于50%的数据：2<=mi<=50,1<=n<=50

对于100%的数据：2<=mi<=100,1<=n<=100,1<=k<=10000,每个积木的重量为不超过100的正整数，所有mi的积大于等于k。本题不卡常。

---

【分析】
这就是个分组背包的变式题…

---

【代码】

//洛谷月赛 1.Y的积木 
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
int n,m,k;
int wei[105][105],sum[105];
int dp[105][10005];
int main()
{
    int i,j,p;
    scanf("%d%d",&n,&k);
    fo(i,1,n)
    {
        int mx=0;
        scanf("%d",&wei[i][0]);
        fo(j,1,wei[i][0])
        {
            scanf("%d",&wei[i][j]);
            mx=max(mx,wei[i][j]);
        }
        sum[i]=sum[i-1]+mx;
    }
    dp[0][0]=1;
    fo(i,1,n)
    {
        int tmp=sum[i];
        for(j=tmp;j;j--)
          fo(p,1,wei[i][0])
          {
              if(j<wei[i][p]) break;
              dp[i][j]+=dp[i-1][j-wei[i][p]];
          }
    }
    for(i=1;i<=10000 && k;i++)
      while(dp[n][i] && k)
        printf("%d ",i),k--,dp[n][i]--;
    return 0;
}