HDU 4020 Ads Proposal 简单的模拟

题意：给你n个广告商，他们有一些广告，每个广告都有相应的点击率和长度，现在为你每个广告商的点击率排名前k个的广告的和。

想法：数据不能每查询一次就找一次，那样必定超时，所以在查询之前要进行预处理。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,k;
struct node
{
	int id,click;
	__int64 len;
}e[500000+5];
int maxx;
__int64 pp[100000+5];
int mark[100000+5];
bool cmp(node a,node b)
{
	if(a.id==b.id) return a.click>b.click;
	else return a.id<b.id;
}
void Input()
{
	scanf("%d%d%d",&n,&m,&k);
	maxx=-1;
	memset(mark,0,sizeof(mark));
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%I64d",&e[i].id,&e[i].click,&e[i].len);
		mark[e[i].id]++;
	}
	for(int i=1;i<=n;i++)
	{
		if(maxx<mark[i])
		{
			maxx=mark[i];
		}
	}
} 
void pretreatment(int ca)
{
	int cnt=1;
	sort(e+1,e+m+1,cmp);
	memset(pp,0,sizeof(pp));
	pp[1]=e[1].len;
	for(int i=2;i<=m;i++)
	{
		if(e[i].id!=e[i-1].id)
		{
			cnt=0;
		}
		pp[++cnt]+=e[i].len;
	}
	for(int i=2;i<=maxx;i++)
	{
		 pp[i]=pp[i-1]+pp[i];
	}
	printf("Case #%d:\n",ca);
	for(int i=1;i<=k;i++)
	{
		int p;
		scanf("%d",&p);
		if(maxx<p) p=maxx;
		printf("%I64d\n",pp[p]);
	}
}
int main()
{
	int t,ca=1;
	scanf("%d",&t);
	while(t--)
	{
		Input();
		pretreatment(ca++);
	}
	return 0;
}