SGU 549 - Dumbbells

来源：互联网发布：微信相关软件编辑：程序博客网时间：2024/05/18 03:58

题目地址： http://acm.sgu.ru/problem.php?contest=0&problem=549

比较好的贪心。

根据题目给的优先级排序，数量多的放前面。

数量排在前k位的maze，假设数量最少的为len个，那么统计每个maze的前len大cost之和。

将统计出来的所有数从大到小排序，然后取前k大。

#include<iostream>#include<cstdio>#include<algorithm>#include<cmath>#include<cstring>using namespace std;int sum[5000],mx[5000];int num[5000];struct Node{int x,y;}s[5000];int cmp(Node a,Node b){if(num[b.x]!=num[a.x]){return num[a.x]>num[b.x];}return a.x==b.x?a.y>b.y:a.x>b.x;}int main(){int n,k;int i,j; int res,cost,len,cnt;while(~scanf("%d%d",&n,&k)){for(i=0;i<=4000;i++){num[i]=0;}for(i=0;i<n;i++){scanf("%d%d",&s[i].x,&s[i].y);num[s[i].x]++;}sort(s,s+n,cmp);j=1;for(i=1;i<n;i++){if(s[i].x!=s[i-1].x){j++;if(j>k && num[s[i].x]<num[s[i-1].x]){break;}}}len=num[s[i-1].x];if(j<k){printf("%d %d\n",0,0);continue;}cnt=0;for(i=0;num[s[i].x]>=len;i+=num[s[i].x]){sum[cnt]=0;for(j=0;j<len;j++){sum[cnt]+=s[i+j].y;}cnt++;}sort(sum,sum+cnt);cost=0;//cout<<cnt<<endl;for(i=cnt-1;i>=0 && i>=cnt-k;i--){cost+=sum[i];} printf("%d %d\n",len,cost);}return 0;}