很多个较小的数字可以随机组合成较大的数字,所以B数组从小到大开始遍历,除了空集,最小的那个存在的个数对应的数字必然是a数组中的数字。
每求出这一部分之后,更新后续的B序列。
分析完后,主要的难点就是怎么去让已求出来的A序列随机组合,更新后续的B序列直接减就可以了。看成01背包问题,让m为背包去装 i,初始值为dp[0] = 1,由于i依次增大,A子集随机组合不会重复。
#include <bits/stdc++.h>
using namespace std;
// #define test TEST
typedef long long ll;
const int maxn=1e4+5;
int cnt[maxn],a[100],dp[maxn];
bool cmp(const int &a,const int &b){
return a>b;
}
int main(int argc, char const *argv[])
{
int T;
scanf("%d",&T);
while(T--) {
int n,m;
scanf("%d %d",&n,&m);
for(int i=0;i<=m;i++){
scanf("%d",&cnt[i]);
}
memset(dp,0,sizeof(dp));
dp[0]=1;//dp数组记录A序列中值为i的次数
int pos=0;
for(int i=1;i<=m;i++){
int temp=cnt[i]-dp[i];//A序列中值为i的个数
for(int j=1;j<=temp;j++){
a[pos++]=i;//对A序列赋值
for(int k=m;k>=i;k--){
dp[k]+=dp[k-i];//和为k的A子集个数相加去更新B序列
}
}
}
for(int i=0;i<n;i++){
printf("%d%c",a[i],i==n-1?'\n':' ');
}
}
return 0;
}