CodeForces ~ 982B ~ Bus of Characters (栈模拟)

题意:公交上有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;
}


猜你喜欢

转载自blog.csdn.net/zscdst/article/details/80484906