hdu6092(dp)

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=6092

题意: 输入格式为, 对于每组测试样例第一行输入两个数 n, m, 接下来一行输入B数组, 有 m + 1个数. 其中 n 表示要构造的数组 A 的长度为 n. m 表示 A 数组的元素和为 m. 第二行的m + 1 个数字表示 A 数组 有子集中和为 0, 1, ..., m 的子集个数.

思路: 可以从 1 到 m 依次确定每个数出现的次数. 用 dp[i] 表示当前答案集中和为 i 的子集数.

枚举 1 <= i <= m, 对于当前 i, 其在 A 中出现的次数为 B[i] - dp[i], 每往 A 中添加一个元素更新一次 dp 数组.

更新 dp 的过程可以看作一个 01 背包过程.

代码:

 #include <iostream>
 #include <stdio.h>
 #include <string.h>
 using namespace std;
 
 const int MAXN = 1e4 + ;
 int dp[MAXN], a[MAXN], sol[MAXN];//dp[i]存储当前答案集中和为i的子集有多少
 
 int main(void){
     int t, n, k;
     scanf("%d", &t);
     while(t--){
         int indx = ;
         memset(dp, , sizeof(dp));
         scanf("%d%d", &n, &k);
         for(int i = ; i <= k; i++){
             scanf("%d", &a[i]);
         }
         dp[] = ;
         for(int i = ; i <= k; i++){
             int cnt = a[i] - dp[i];
             for(int j = ; j < cnt; j++){
                 sol[indx++] = i;
                 for(int l = k; l >= i; l--){//注意先更新小的数会影响后面的更新
                     dp[l] += dp[l - i];
                 }
             }
         }
         for(int i = ; i < indx; i++){
             if(i) printf(" ");
             printf("%d", sol[i]);
         }
         puts("");
     }
     return ;
 }