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;
}