莫名TLE,莫名WA,AC好难。。。
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream>
#include<string>
#include <set>
//a&3==a%4
using namespace std;
#define ll long long
#define mem(a) memset(a,0,sizeof(a))
const int eps=1e-8;
const int maxn=10010;//须填写
const int inf=0x3f3f3f3f;
int a[maxn];
int b[maxn];
int dp[maxn];
int c[maxn];
int main()
{
int kase;
int n,m;
scanf("%d",&kase);
while(kase--)
{
mem(a);
mem(b);
mem(dp);
dp[0]=1;
int p=0;
scanf("%d%d",&n,&m);
for(int i=0;i<=m;i++)
scanf("%d",&b[i]);
for(int i=1;i<=m;i++)
{
c[i]=b[i]-dp[i];//记录有多少个相同元素,即要取多少次
for(int j=0;j<c[i];j++)
{
/*if(p==n)//已经找出a中所有元素
break;*/
a[p++]=i;//将这个元素加入a中
for(int k=m;k>=i;k--)/*将这个元素从后边包含这个元素的树木中减去,先把需要减去的数记下来然后一起在36行减去*/
{
dp[k]+=dp[k-i];
}
}
}
printf("%d",a[0]);//输出
for(int i=1;i<p;i++)
{
printf(" %d",a[i]);
}
cout<<endl;
}
return 0;
}