思路
简单模拟:
根据题意,我们知道,每一次外向的人上车的时候,一定有内向的人在,所以内向人的位置中宽度最大的就是外向人选的位置,那么只要我们每一次把内向人的位置记下来,每次遇到外向人的时候,取其中的宽度最大的值就可以了,
内向的人每次选的都是没有人的座位,且选其中宽度最小的,那么我们先把每个row按照宽度从小到大排序,只要遇到一个内向的人,就取idx上的元素,之后idx++即可,然后把这个座位的编号存在stack中,来给外向的人用,
因为入栈的的顺序宽度是从小到大的,所以栈顶的元素就是有人且宽度最大的位置,直接给外向人就可以了。
代码
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <stack>
#include <cmath>
using namespace std;
const int N=200010;
pair<int,int> a[N];
bool cmp(pair<int,int> a,pair<int,int> b){
return a.first<b.first;
}
int main(){
int n;
cin >> n ;
for(int i=1;i<=n;i++){
int x;
cin >> x;
a[i]={x,i};
}
sort(a+1,a+n+1,cmp);
stack<int> sta;
int idx=1;
for(int i=0;i<2*n;i++){
char c;
cin >> c;
if(c=='1'){
int t=sta.top();
sta.pop();
cout << t << " ";
}
else{
sta.push((a[idx].second));
cout << a[idx++].second << " ";
}
}
}