例题6-2 UVA514 Rails（19行AC代码）

紫书刷题进行中，题解系列【GitHub|CSDN】

例题6-2 UVA514 Rails（19行AC代码）

题目大意

给定包含n个数字的序列A，问数字序列B（1~n顺序排列）能否通过一个栈改变顺序来得到A。通过栈改变顺序时不可出现穿越数字。

思路分析

很自然的用栈来模拟整个过程，用j记录目标序列a和模拟序列连续相同个数，枚举1-n，先压入栈中，若栈顶等于a[j]，则退栈，并且j++，继续比较下一个栈顶。

若最后栈为空，表示合法；否则，非法

AC代码（C++11，栈妙用）

#include<bits/stdc++.h>
using namespace std;
int n, a[1010];
int main() {
    while (cin >>n && n != 0) {
        while (cin >>a[1] && a[1] != 0) {
            for (int i = 2; i <= n; i ++) scanf("%d", &a[i]); // 输入
            stack<int> stk; // 模拟栈
            int j = 1; // 判断联系满足的个数
            for (int i = 1; i <= n; i ++) { // 1-n依次压入
                stk.push(i); // 压栈
                while (!stk.empty() && stk.top() == a[j]) stk.pop(), j ++; // 当栈非空且栈顶为a[j]时，出栈
            }
            printf("%s\n", stk.empty() && j == n+1 ? "Yes" : "No"); // 栈空表示合法
        }
        puts(""); // 两组间空行
    }
    return 0;
}