Description
n场考试,第i场考试会从c[i]道题目中等概率选出一题,所有题目都不同,现在只能学会k道题,问如果安排才能使得通过的考试的数量的期望最大
Input
第一行两个整数k和n表示考试数量和能学会的题数,之后n个整数c[i]表示第i场考试题目数量
(0<=k < 200000,1<=n<=2000,1<=c[i]<=100,k < sum{c[i]})
Output
输出n个整数表示每场考试要学习的题目数,使得通过的考试数量的期望最大
Sample Input
2 2
3 2
Sample Output
0 2
Solution
贪心,假设每场考试学会d[i]道题,那么通过的考试数量的期望就是sum{d[i]/c[i]},而sum{d[i]}=k,为使期望最大,那么肯定先满足小的分母再顾及大的分母,所以给c[i]排个序从小往大扫即可
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 2222
int k,n;
struct node
{
int c,id;
bool operator<(const node& b)const
{
return c<b.c;
}
}a[maxn];
int ans[maxn];
int main()
{
freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
while(~scanf("%d%d",&k,&n))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i].c);
a[i].id=i;
}
sort(a+1,a+n+1);
for(int i=1;i<=n;i++)
{
if(a[i].c<=k)
{
k-=a[i].c;
ans[a[i].id]=a[i].c;
}
else
{
ans[a[i].id]=k;
k=0;
}
}
for(int i=1;i<=n;i++)
printf("%d%c",ans[i],i==n?'\n':' ');
}
}