题目大意是有一批学生上大学,但是需要一些奖学金才上得起,学校给的奖学金aid有限,并且学校的教室只能让N个学生来上学,N为奇数,所以学校希望在在所有参加入学考试的C名学生中选取N个学生,为了提高教学质量,在满足这N个学生的奖学金的同时,希望这N个学生的成绩中位数最高。给定一些学生的成绩和需要的奖学金,以及学校最多能支付的奖学金,输出最大的中位数,如果不能满足任何一种选法,就输出-1。
首先对C名学生按照成绩进行排序,因为题目需要的是输出满足条件的最大中位数,即当选定一个学生作为中位数时,只需要在大于该成级的和小于该成绩的学生中分别选取需要奖学金总和最小的N/2学生。到这里,就转化成为了在N个数中选取TOPK的问题了,利用堆的话复杂度是nlog(k)。
#ifndef HEAD
#include <stdio.h>
#include <vector>
#include <math.h>
#include <string.h>
#include <string>
#include <iostream>
#include <queue>
#include <list>
#include <algorithm>
#include <stack>
#include <map>
using namespace std;
#endif // !HEAD
struct MUFA
{
int score;
int aid;
};
int curaids[2][100001];
MUFA mufas[100001];
int mufscomp(const void* a1, const void* a2)
{
return ((MUFA*)a1)->score - ((MUFA*)a2)->score;
}
int main()
{
#ifdef _DEBUG
freopen("d:\\in.txt", "r", stdin);
#endif
int N, C, F;
memset(curaids, 0, sizeof(curaids));
scanf("%d %d %d\n", &N, &C, &F);
for (size_t i = 0; i < C; i++)
{
scanf("%d %d\n", &mufas[i].score, &mufas[i].aid);
}
qsort(mufas, C, sizeof(MUFA), mufscomp);
int start = N / 2;
int end = C - start - 1;
priority_queue<int, vector<int>> aids,aids1;
int curaid = 0;
for (int i = C - 1; i > end;i--)
{
curaid += mufas[i].aid;
aids.push(mufas[i].aid);
}
curaids[0][end] = curaid;
for (int i = end -1; i >= start; i--)
{
if (mufas[i + 1].aid < aids.top())
{
curaids[0][i] = curaids[0][i + 1] - aids.top() + mufas[i + 1].aid;
aids.pop();
aids.push(mufas[i + 1].aid);
}
else
curaids[0][i] = curaids[0][i + 1];
}
curaid = 0;
for (int i = 0; i < start;i++)
{
curaid += mufas[i].aid;
aids1.push(mufas[i].aid);
}
curaids[1][start] = curaid;
for (int i = start + 1; i <= end;i++)
{
if (mufas[i-1].aid < aids1.top())
{
curaids[1][i] = curaids[1][i - 1] - aids1.top() + mufas[i - 1].aid;
aids1.pop();
aids1.push(mufas[i - 1].aid);
}
else
{
curaids[1][i] = curaids[1][i - 1];
}
}
int res = -1;
for (int i = end; i >= start;i--)
{
if (mufas[i].aid + curaids[0][i] + curaids[1][i] <= F)
{
res = mufas[i].score;
break;
}
}
printf("%d\n", res);
return 0;
}