Atcoder beginner contest 123D Cake 123

https://atcoder.jp/contests/abc123/tasks/abc123_d
题意:3种型号的蛋糕分别x,y,z个,每个的美味值已知,3种型号蛋糕各取一个,总美味值等于3个蛋糕的美味值乘积。求所有总美味值组成情况的前k个。
思路:打无脑暴力肯定不行,只好打有脑暴力。先暴力算出A和B中的所有可能,取前k个,再用这k个跟C中暴力求所有可能,再取前k个。

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define ll long long
#define pii pair
#define mp make_pair
#define fi first
#define se second
#define inf 0x7fffffff
using namespace std;
ll a[1010],b[1010],c[1010];
vector<ll> v,v1;
int main()
{
	int i,j,k,n,m,x,y,z,w;
	scanf("%d%d%d%d",&x,&y,&z,&w);
	for(i=0;i<x;i++)
	{
		scanf("%lld",&a[i]);
	}
	for(i=0;i<y;i++)
	{
		scanf("%lld",&b[i]);
	}
	for(i=0;i<z;i++)
	{
		scanf("%lld",&c[i]);
	}
	for(i=0;i<x;i++)
	{
		for(j=0;j<y;j++)
		{
			v.push_back(a[i]+b[j]);
		}
	}
	sort(v.begin(),v.end());
	reverse(v.begin(),v.end());
	v.resize(w);
	for(i=0;i<v.size();i++)
	{
		for(j=0;j<z;j++)
		{
			v1.push_back(v[i]+c[j]);
		}
	}
	sort(v1.begin(),v1.end());
	reverse(v1.begin(),v1.end());
	for(i=0;i<w;i++)
	{
		printf("%lld\n",v1[i]);
	//	cout<
	}
	return 0;
}

你可能感兴趣的:(atcoder)