题目大意
有n个数,任意组合成k项相加,输出所有可能结果。
抖机灵
如果dp,我们发现状态数很多,都是因为需要保留选了多少项。
能不能让限制不要那么严格呢?就是说,一定要恰好k项,如果k项以内的都算,那不是就简单很多了吗?
想让限制变松,我们需要其中一个数为0。没有0,我们就造一个。找到最小的数mi,然后所有数都减去mi,最后输出时加上mi*k即可。
于是很好做了,不超过k步能得到的所有都合法,bfs即可。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=500+10;
int dl[maxn*maxn],step[maxn*maxn],a[maxn],s[20];
bool bz[maxn*maxn];
int i,j,k,l,r,t,n,m,mi,mx,top;
void write(int x){
top=0;
while (x){
s[++top]=x%10;
x/=10;
}
while (top){
putchar(s[top]+'0');
top--;
}
putchar(' ');
}
int main(){
scanf("%d%d",&n,&m);
fo(i,1,n) scanf("%d",&a[i]);
sort(a+1,a+n+1);
mi=a[1];
j=0;
fo(i,1,n) a[i]-=mi;
mx=a[n]*m;
step[1]=dl[r=1]=l=0;
bz[0]=1;
while (l<r){
k=dl[++l];
if (step[l]==m) continue;
fo(i,1,n)
if (!bz[k+a[i]]){
bz[k+a[i]]=1;
dl[++r]=k+a[i];
step[r]=step[l]+1;
}
}
fo(i,0,mx)
if (bz[i]) write(i+mi*m);
}