水题--PAT.1051. [栈]

判断一个序列是否是某个栈的弹出顺序的步骤是
(1)若下一个要弹出的数字刚好在栈顶,则弹出该数字;
(2)若下一个要弹出的数不在栈顶,则不断压入数据直到压入要弹出的数据;
(3)若所有的数据都已压入栈内,但是下一个要弹出的数字不在栈顶,
则该序列不可能是该栈的一个弹出序列

/**************************
//@Author: 3stone
//@ACM: PAT-A1051   
//@Time: 18/2/2
//@IDE: VS2017
***************************/
#include<cstdio>
#include<iostream>
#include<vector>
#include<stack>
using namespace std;

int main() {
    int M, N, K, temp;
    vector<int> seq;
    while (scanf("%d %d %d", &M, &N, &K) != EOF) {
        for (int i = 0; i < K; i++) {
            seq.clear(); //清空序列
            //input
            for (int j = 0; j < N; j++) {
                scanf("%d", &temp);
                seq.push_back(temp);
            }

            stack<int> S;
            for (int k = 1, cur = 0; k <= N && cur < N; k++) { //模拟入栈
                S.push(k);
                if (S.size() > M) //每次压入时判断是否爆栈
                    break;
                while (!S.empty() && seq[cur] == S.top()) {//栈顶匹配
                    cur++;
                    S.pop(); //出栈

                }
            }

            if (S.empty()) printf("YES\n");
            else printf("NO\n");

        }
    }//while
    system("pause");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_26398495/article/details/79244220