POJ 1363 Rails

解题思路:

本题是一道模拟,我们先构造一个栈,将1~n入栈,并用一个标记来表示当前比较到了所给序列的那一个元素.如果要入栈的元素与我们当前所标记的元素相同,则比较栈顶与我们下一个要标记到的元素,直到栈为空或这两个元素不相同为止,更新标记,接着循环直到1~n全部入栈.如果栈为空,则说明"Yes",如果不为空,我们就比较栈顶和标记元素,如果相同,就弹出栈顶;最后如果能做到栈为空则说明"Yes",否则为"No".

AC代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <stack>

using namespace std;
const int N = 1e5+5;

int n, a[N];

bool solve() {
    stack<int> st;
    int p = 1;//p为标记 
    for(int i=1; i<=n; i++) {
        while(true) {
            if(p <= n && a[i] == p) {
                p++;
                break;
            }
            else if(!st.empty() && a[i] == st.top()) {
                st.pop();
                break;
            }
            else {
                if(p <= n) st.push(p++);
                else return false;
            }
        }
    }
    return true;
}

int main() {
    while(cin >> n && n) {
        while(true) {
            cin >> a[1];
            if(a[1] == 0) break;
            for(int i=2; i<=n; i++) cin >> a[i];
            cout << (solve() ? "Yes" : "No") << "
";
        }
        puts("");
    }
}