SGU 549 – Dumbbells

 

 

題目地址： 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;}