优先队列--堆排序思想

核心

小根堆放大数，大根堆放小数

priority_queue<int,vector<int>,less<int>>q1;//从大到小---大根堆
priority_queue<int,vector<int>,greater<int>>q2;//从小到大--小根堆

HDU3182

题目链接：https://vjudge.net/contest/450708#problem/C

给定一个长为n的序列，请你求出每当我们考虑前k(k=1；k<=n；k+=2)个数字时中位数是多少？

思路：

priority_queue<int> p1;     //大根堆，存储小于等于中位数的数，p1.top()为中位数
 
priority_queue<int,vector<int>,greater<int> > p2;  //小根堆，存储大于中位数的数

以大根堆的堆顶作为分界点，小于等于的就加到小根堆里，大于时就加入到大根堆里。同时我们还需要保证两边堆里的数据个数相差不能大于1个，始终是q1.size()==q2.size()+1。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int maxx=1e5+10;
typedef long long ll;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int a,n,x;
        cin>>a>>n;
         priority_queue<int,vector<int>,less<int>>q1;//从大到小
         priority_queue<int,vector<int>,greater<int>>q2;//从小到大
         vector<int>ans;
        for(int i=1;i<=n;i++)
        {
            cin>>x;
            if(q1.empty())
                q1.push(x);
            else
            {
                if(x>q1.top())
                {
                    q2.push(x);
                }
                else
                {
                     q1.push(x);
                }
            }

            if(q1.size()<q2.size())
            {
                q1.push(q2.top());
                q2.pop();
            }
            if(q2.size()<q1.size()-1)
            {
                q2.push(q1.top());
                q1.pop();
            }

            if(i%2!=0)
            {
                ans.push_back(q1.top());
            }
        }
        cout<<a<<" "<<n/2+1<<endl;
        for(int i=1;i<=ans.size();i++)
        {
            if(i%10!=1)
            {
                cout<<" ";
            }
            cout<<ans[i-1];
            if(i%10==0)
            {
                cout<<endl;
            }
        }
        cout<<endl;
    }
}

 

POJ1442

题目链接：https://vjudge.net/contest/450928#problem/F

题意：提供一个数组，要求你输出在u[i]范围内第i小的数字

思路：对顶堆想法，把数组进行分隔存储到两个堆里，因为需要的是小的数字，所以每一次进行数据分堆时，保证小根堆堆顶的数据就是我们寻找的数据即可，其次题目保证了，数据范围是递增的，不需要担心数组还会缩回去这个问题。

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxx=3e4+10;
int a[maxx];
int main()
{
    int n,m;
    priority_queue<int,vector<int>,greater<int>>q1;//从小到大
    priority_queue<int,vector<int>,less<int>>q2;
    scanf("%d%d",&m,&n);
    for(int i=0;i<m;i++)
        scanf("%d",&a[i]);
    int u,j=0,tmp;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&u);
        while(j<u)
        {
            q1.push(a[j]);
            if(!q2.empty()&&q1.top()<q2.top())
            {
                tmp=q1.top();
                q1.pop();
                q1.push(q2.top());
                q2.pop();
                q2.push(tmp);
            }
            j++;
        }
        printf("%d\n",q1.top());
        q2.push(q1.top());
        q1.pop();
    }

}