题目传送门
大意
在一辆公交车中有 n n n 排座位,第 i i i 排两个座位的宽度均为 w i w_i wi。所有的 w i w_i wi 互不相同。
初始时,公交车是空的。接下来会依次停靠 2 n 2n 2n 个站,每一站将上来一名乘客。乘客分
为两类:
-
内向者:此类乘客总是会选择两个座位都是空的那一排就坐,如果有多排都是空的,他将会选择 w i w_i wi 最小的那一排中任意一个空座坐下。
-
外向者:此类乘客总是会选择已有一人就坐(当然是内向者)的那一排,如果有多排都满足条件,他会选择 w i w_i wi 最大的那一排的空座坐下。
现在给定每一排的宽度 w i w_i wi 以及乘客上车的顺序。请确定每一个乘客将会选择哪一排坐
下。
思路
这题我们可以用两个堆,一个大根堆和一个小根堆,分别用来表示内向者的首选和外向
者的首选座位。
因为一开始座位都是空的,所以全部的座位都会分给内向者,每次选座位时,堆顶人
数加一。若当前剩下一个座位,就将堆顶弹出。
代码
#include<bits/stdc++.h>
using namespace std;
int n,x;
char s[400005];
priority_queue<int,vector<int>,greater<int> > q1;
priority_queue<int> q2;
map<int,int> mp;
int main(){
cin>>n>>s+1;
for(int i=1;i<=n;i++){
cin>>x;
mp[x]=i;
q1.push(x);
}
for(int i=1;i<=n*2;i++){
if(s[i]=='0'&&!q1.empty()){
x=q1.top();
q1.pop();
q2.push(x);
}else if(!q2.empty()){
x=q2.top();
q2.pop();
}
cout<<mp[x]<<" ";
}
return 0;
}