openjudge 牛的选举

题目：

3:牛的选举
查看 提交 统计 提问
总时间限制: 1000ms 内存限制: 65536kB
描述
现在有N（1<=N<=50000）头牛在选举它们的总统，选举包括两轮：第一轮投票选举出票数最多的K（1<=K<=N）头牛进入第二轮；第二轮对K头牛重新投票，票数最多的牛当选为总统。



现在给出每头牛i在第一轮期望获得的票数Ai（1<=Ai<=1,000,000,000），以及在第二轮中（假设它进入第二轮）期望获得的票数Bi（1<=Bi<=1,000,000,000），请你预测一下哪头牛将当选总统。幸运的是，每轮投票都不会出现票数相同的情况。


输入
第1行：N和K
第2至N+1行：第i+1行包括两个数字：Ai和Bi
输出
当选总统的牛的编号（牛的编号从1开始）
样例输入
5 3
3 10
9 2
5 6
8 4
6 5
样例输出
5

===========================================================================

还是基本的排序问题，数据量有点大，所以还要稍加注意算法的时间复杂度；这里为了练习，写了一个快速排序，一个最大堆。

代码清单：

#include <iostream>
#include <cstdio>
using namespace std;

#define MAXN 50005

typedef struct _ballot
{
	unsigned int first;
	unsigned int second;
	unsigned int id;
} ballot;

ballot cand[MAXN];
ballot MaxHeap[MAXN];

//按第一轮得票数排序:快速排序。注意关键字是从大到小排列。
void QuickSortbyFirst(ballot a[], int m, int n)
{
	int i=m;
	int j=n;
	int pivot=a[(m+n)/2].first;

	do 
	{
		while(a[i].first>pivot && i<n)	++i;
		while(a[j].first<pivot && j>m)	--j;

		if (i<=j)
		{
			cand[MAXN-1].first=a[i].first;
			cand[MAXN-1].second=a[i].second;
			cand[MAXN-1].id=a[i].id;

			a[i].first=a[j].first;
			a[i].second=a[j].second;
			a[i].id=a[j].id;

			a[j].first=cand[MAXN-1].first;
			a[j].second=cand[MAXN-1].second;
			a[j].id=cand[MAXN-1].id;
			
			++i;
			--j;
		}
	} while (i<=j);

	if(i<n)	QuickSortbyFirst(a, i, n);
	if(m<j)	QuickSortbyFirst(a, m, j);
}

//最大堆的siftup操作，以第二轮得票数为指标
void SiftUpbySecond(int pos)
{
	int child=pos;
	int father=(child-1)/2;

	//挖坑
	cand[MAXN-1].first=MaxHeap[child].first;
	cand[MAXN-1].second=MaxHeap[child].second;
	cand[MAXN-1].id=MaxHeap[child].id;

	while (father>=0 && child>=1)
	{
		if (cand[MAXN-1].second > MaxHeap[father].second)
		{
			MaxHeap[child].first=MaxHeap[father].first;
			MaxHeap[child].second=MaxHeap[father].second;
			MaxHeap[child].id=MaxHeap[father].id;

			child=father;
			father=(child-1)/2;
		}
		else
			break;
	}

	//填坑
	MaxHeap[child].first=cand[MAXN-1].first;
	MaxHeap[child].second=cand[MAXN-1].second;
	MaxHeap[child].id=cand[MAXN-1].id;
}

void BuildMaxHeap(int k)
{
	for (int i=0; i<k; ++i)
	{
		MaxHeap[i].first=cand[i].first;
		MaxHeap[i].second=cand[i].second;
		MaxHeap[i].id=cand[i].id;

		SiftUpbySecond(i);
	}
}

int main()
{
	freopen("D:\\in.txt", "r", stdin);
	freopen("D:\\out.txt", "w", stdout);	

	int n, k;
	scanf("%d%d", &n, &k);
	for (int i=0; i<n; ++i)
	{
		scanf("%d%d", &(cand[i].first), &(cand[i].second));
		cand[i].id=i+1;
	}

	QuickSortbyFirst(cand, 0, n-1);

	BuildMaxHeap(k);

	printf("%d", MaxHeap[0].id);

	return 0;
}