3197999_re.cc

来自「北大大牛代码 1240道题的原代码 超级权威」· CC 代码 · 共 63 行

#include <stdio.h>
#include <algorithm>

using namespace std;

int num[101][2001];
int ans[10000], cnt;
int m, n, t;

int power(int a,int b)
{
	if(b==1)
		return a;
	return a*power(a,b-1);
}



void solve(int p,int s)
{
	int i;

	if(p==m)
	{
		ans[cnt++] = s;
		return ;
	}
	for(i = 0; i <t; i++)
	{
		solve(p+1,s+num[p][i]);
	}
}

int main()
{
	int cas;
	int i, j;

	scanf("%d",&cas);
	while(cas-- > 0)
	{
		scanf("%d%d",&m,&n);
		for(i = 0; i < m; i++)
		{
			for(j = 0; j < n; j++)
			{
				scanf("%d",&num[i][j]);
			}
			sort(num[i],num[i]+n);
		}
		for(t = 1; power(t,m) < n ; t++);
		cnt = 0;
		solve(0,0);
		sort(ans,ans+cnt);
		for(i = 0; i < n; i++)
		{
			printf("%d ",ans[i]);
		}
		printf("\n");
	}
	return 0;
}