题意:公交上有N排凳子,每排有两个凳子,每一排的凳子宽度不一样。有一些内向和外向的人按照顺序上车。
外向的人(0):只会选择没人的一排坐下,如果有很多排符合要求,他会选择座位宽度最小的坐下。
内向的人(1):只会选择有人的一排坐下,如果有很多排符合要求,他会选择座位宽度最大的坐下。
输入N,然后输入N个整数表示每一排的的凳子长度,然后输入一个长度为2*N的01串表示2*N个人。从1~2*N输出这些人的所在的排,序列保证合法。
思路:可以发现内向的人和外向的人刚好相反,一个会选择座位宽度最小的,一个会选择有人且座位宽度最大的。很符合栈的性质,所以我们用一个栈模拟,先按照宽度从小到大排序,如果来了一个外向的人,那么就坐到一个没人且宽度最小的排,把这个排的ID入栈,如果来了一个外向的人,那么就坐到栈顶那一排,栈顶出栈。
#include<bits/stdc++.h> using namespace std; const int MAXN = 200005; int n; pair<int, int> a[MAXN]; char str[MAXN*2]; stack<int> S; int main() { scanf("%d", &n); for (int i = 0; i < n; i++) { scanf("%d", &a[i].first); a[i].second = i+1; } sort(a, a+n); scanf("%s", str); int cnt = 0; for (int i = 0; i < 2*n; i++) { if (str[i] == '0') { printf("%d ", a[cnt].second); S.push(a[cnt].second); cnt++; } if (str[i] == '1') { printf("%d ", S.top()); S.pop(); } } return 0; }