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