优先队列 - Moo University - Financial Aid - POJ - 2010

优先队列 - Moo University - Financial Aid - POJ - 2010

题意：

$首行输入三个整数n,m,F$

$接着输入m行正整数数对(a_i,b_i)。$

$从m个数对中挑选n个,(a_{i_1},b_{i_1}),(a_{i_2},b_{i_2}),...,(a_{i_n},b_{i_n})，需满足{\sum }_{k=1}^n{b}_{i_k}\le F。$

$在满足以上约束条件的前提下，使得数列a_{i_1},a_{i_2},...,a_{i_n}的中位数最大。输出最大的中位数。$

$保证n是奇数。$

$若无法满足前提条件，输出-1。$

Sample Input：

3 5 70
30 25
50 21
20 20
5 18
35 30

Sample Output：

35

数据范围：

$$\begin{gathered} 1\le n\le 19,999，n\le m\le 100,000，0\le F\le 2,000,000,000 \\ Timelimit:1000ms，Memorylimit:30000kB \end{gathered}$$

分析：

$由于n是奇数，中位数两侧的数的个数相同，都是\lfloor \frac{n}{2}\rfloor 个。$

$因此，我们可以将数对(a_i,b_i)以a_i为第一关键字排序，然后从大到小开始枚举中位数，判断是否可行即可。$

$枚举中位数后a_i后，相当于a_i两侧分别挑选\lfloor \frac{n}{2}\rfloor 个数对。$

$记a_i左侧挑选的\lfloor \frac{n}{2}\rfloor 个数对为(a_{l_1},b_{l_1}),(a_{l_2},b_{l_2}),...,(a_{l_{\frac{n}{2}}},b_{l_{\frac{n}{2}}})，记S_l={\sum }_{k=1}^{\lfloor \frac{n}{2}\rfloor }{b}_{l_k}$

$右侧挑选的\lfloor \frac{n}{2}\rfloor 个数对为(a_{r_1},b_{r_1}),(a_{r_2},b_{r_2}),...,(a_{r_{\frac{n}{2}}},b_{r_{\frac{n}{2}}})，记S_r={\sum }_{k=1}^{\lfloor \frac{n}{2}\rfloor }{b}_{r_k}$

$要使得总和不超过F，至少要满足a_i左右两侧的S_{l_{min}}+S_{r_{min}}+b_i\le F。$

$故我们预处理数组l[i]来存储在[1,i-1]挑选\lfloor \frac{n}{2}\rfloor 个数对的S_{l_{min}}，$

$用r[i]来存储在[i+1,m]挑选\lfloor \frac{n}{2}\rfloor 个数对的S_{r_{min}}。$

$问题转化为求某段区间内前\lfloor \frac{n}{2}\rfloor 小的数的和。$

$这我们可以通过大根堆来维护。$

$最后仅需判断l[i]+r[i]+b[i]\le F是否成立即可。$

代码：

#include<cstring>
#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>

#define P pair<int,int>
#define x first
#define y second

using namespace std;

const int N=1e5+10;

int n,m,F;
P calf[N];
int l[N],r[N];
priority_queue<int> heap;

int main()
{
    cin>>n>>m>>F;
    for(int i=1;i<=m;i++) scanf("%d%d",&calf[i].x,&calf[i].y);
    
    sort(calf+1,calf+m+1);
    
    memset(l,0x3f,sizeof l);
    memset(r,0x3f,sizeof r);
    
    //求l[i]
    int sum=0;
    for(int i=1;i<=m;i++) 
    {
        if(heap.size()<n/2) {heap.push(calf[i].y);sum+=calf[i].y;}
        else
        {
            l[i]=sum;
            if(calf[i].y<heap.top())
            {
                sum-=heap.top();
                heap.pop();
                heap.push(calf[i].y);
                sum+=calf[i].y;
            }
        }
    }
    
    //求r[i]
    heap=priority_queue<int>();
    sum=0;
    for(int i=m;i;i--) 
    {
        if(heap.size()<n/2) {heap.push(calf[i].y);sum+=calf[i].y;}
        else
        {
            r[i]=sum;
            if(calf[i].y<heap.top())
            {
                sum-=heap.top();
                heap.pop();
                heap.push(calf[i].y);
                sum+=calf[i].y;
            }
        }
    }
    
    bool flag=true;
    for(int i=m-n/2;i>n/2;i--)
        if(l[i]+calf[i].y+r[i]<=F)
        {
            flag=false;
            cout<<calf[i].x<<endl;
            break;
        }
    
    if(flag) puts("-1");
    
    return 0;
}

---