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 + 10;
int dp[MAXN], a[MAXN], sol[MAXN];//dp[i]存储当前答案集中和为i的子集有多少

int main(void){
    int t, n, k;
    scanf("%d", &t);
    while(t--){
        int indx = 0;
        memset(dp, 0, sizeof(dp));
        scanf("%d%d", &n, &k);
        for(int i = 0; i <= k; i++){
            scanf("%d", &a[i]);
        }
        dp[0] = 1;
        for(int i = 1; i <= k; i++){
            int cnt = a[i] - dp[i];
            for(int j = 0; j < cnt; j++){
                sol[indx++] = i;
                for(int l = k; l >= i; l--){//注意先更新小的数会影响后面的更新
                    dp[l] += dp[l - i];
                }
            }
        }
        for(int i = 0; i < indx; i++){
            if(i) printf(" ");
            printf("%d", sol[i]);
        }
        puts("");
    }
    return 0;
}