优先队列的两份模板
1.大的数排在前面
#define pii pair<int ,int>
priority_queue< pii,vector<pii> >q;
q.push(pii(a[i],i));
q.top().first//这是第一个数
q.top().second//这是第二个数
2.小的数优先
#define pii pair<int ,int>
priority_queue< pii,vector<pii>,greater<pii> >q;
//其他的都和上面类似
下面来看两道例题,这两道题思路几乎是一模一样的,唯一不同就是一个取大数,一个取小数。
我们直接上代码
#include<bits/stdc++.h>
#define pii pair<int ,int>
using namespace std;
const int N=2e6+6;
int a[N];
int n,k;
priority_queue< pii,vector<pii> >q;
int main( )
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
{
q.push(pii(a[i],i));
if(i>=k)//如果这里的输入数量超过了k,就要开始判断了
{
while(q.top().second<=i-k) q.pop();//整段代码的关键,如果此时的范围已经在其之前,那么就要弹出,重新输入
printf("%d\n",q.top().first);
}
}
return 0;
}
这个是取最小值,如最开头所说的那样,加个greater即可实现
我们先贴上一段跟上面码风一样的代码
#include<bits/stdc++.h>
#define pii pair<int ,int>
using namespace std;
const int N=2e6+6;
int a[N];
int n,k;
priority_queue< pii,vector<pii>,greater<pii> >q;
int main( )
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) cin>>a[i];
cout<<0<<endl;
for(int i=1;i<n;i++) //仔细读这道题就会发现它的最后一个数是一个废数,没有任何意义
{
q.push(pii(a[i],i));//这里细想这个i就是目前队列里存的数的个数
if(i<k)
cout<<q.top().first<<endl;//这个是输入的比k小的时候
else
{
while(q.top().second<=i-k) q.pop();//如果此时的范围已经在其之前,那么就要弹出,重新输入
printf("%d\n",q.top().first);
}
}
return 0;
}
但是很遗憾,这道题由于数据卡的比较严,这样的写法只能得到80分,还有两个点是TLE。然后我又去翻了一下别人的题解,找到了一份这样的答案。
#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
#define pii pair<int,int>
#define val first//元素的值
#define pos second//元素的位置
int n,m;
priority_queue<pii,vector<pii>,greater<pii> > q;//默认以pair的first作为排序关键字
void input(void)
{
scanf("%d%d",&n,&m);
for(register int i=1,a;i<=n;++i)//卡常
{
scanf("%d",&a);
if(i==1)//特判
{
puts("0");
q.push(pii(a,i));//别忘了存入第一个
continue;
}
while(i>m && q.top().pos<i-m)//i>m才进入,接着才判断是否在这个区间内。
q.pop();
printf("%d\n",q.top().val);
q.push(pii(a,i));//区间左闭右开,则先输出再入队
}
}
int main()
{
input();
return 0;
}
其实主体思路大同小异,他这样写之所以可以通过全部案例是因为这里没有一开始将所有的数据全部压进优先队列中,而是边处理输出,边进行压入操作,这样的话比较而言就比较省时间和空间。
对于这些卡数据的题,也要有这种思路,就是边输出,边处理的思想。