poj2010,无语的题,真的做不来,以后见到这类型再说吧,我用的堆(二分也可做)

来源:互联网 发布:淘宝福袋怎么设置 编辑:程序博客网 时间:2024/06/04 15:19

题意: 有 c 头牛考上了大学,这大学免费给牛学费,但总的学费最多只有 f ,因此只有一些牛能去, 学校决定录取 n 头牛,并且学校录取 n 头牛后牛的分数的中位数尽量大,若不能录取,则输出 -1;

想法:开始很天真的直接用堆来做,但发现直接WR。。

方法:是先处理整个排了序的数组;
因为要找中位数,所以对它起影响的只有它本身;
因此,我们直接将前面的 n/2 个值算出来;
再把后面的 n/2 个值算出来;
再从大到小枚举中位数,找到的第一个就是答案;
这儿算前面和后面就需要用到堆来维护;
即用大顶堆维护最小和。
为什么是最小和,再想想吧,两边最小之后。。。

代码如下:

#include <cstdio>#include <iostream>#include <cstring>#include <cmath>#include <string>#include <algorithm>#include <queue>#include <map>#include <set>#include <vector>#define maxn 100100#define INF 1e7#define ll long long#define x first#define y secondusing namespace std;typedef pair<int, ll> P;P C[maxn];ll low[maxn], up[maxn];int main(){    int n, c;    ll f;    while(~scanf("%d%d%lld", &n, &c, &f))    {        for (int i = 0; i < c; ++i)        {            scanf("%d%lld", &C[i].x, &C[i].y);        }        sort(C, C + c);        priority_queue<ll> que;        ll sum = 0;        for (int i = 0; i < n/2; ++i)  //将前面的n/2个数直接算出来        {            sum += C[i].y;            que.push(C[i].y);            low[i] = 0x3f3f3f3f * 2;        }        for (int i = n/2; i < c; ++i) //枚举算出每一位上的最小值        {            low[i] = sum;            sum += C[i].y;            que.push(C[i].y);            sum -= que.top();            que.pop();        }        while (!que.empty()) que.pop();        sum = 0;        for (int i = c - 1; i >= c - n/2; --i)        {            sum += C[i].y;            que.push(C[i].y);            up[i] = 0x3f3f3f3f * 2;        }        for (int i = c - n/2 - 1; i >= 0; --i)        {            up[i] = sum;            sum += C[i].y;            que.push(C[i].y);            sum -= que.top();            que.pop();        }        int flag = 0;        for (int i = c - 1; i >= 0; --i)        {            if (low[i] + C[i].y + up[i] <= f)            {                flag  = 1;                printf("%d\n", C[i].x);                break;            }        }        if (!flag) printf("-1\n"); //若没有符合条件则返回-1    }    return 0;}
0 0
原创粉丝点击