栈的简单应用(模拟出栈进栈)

题目描述:

输入格式:
输入有两行,第一行是一个正整数 N (1 <= N <= 100),表示输入魔法机序列的长度,第二行是序列 a,共有 N 个整数,表示要得到的目标序列。序列为 1 到 N 的排列,即序列 a 长度为 N,保证序列中的整数都不相同,且整数在区间 [1, N] 内。
输出格式:
输出一行。如果能通过魔法机得到序列 a,则输出“YES”,否则输出“NO”。
输入:
5
2 1 5 4 3
输出:
YES


这道题是栈的入门题,直接用栈模拟即可。
我们首先对样例进行分析:
数字 1 进栈,将栈顶元素与输出序列首元素比较,发现两者不等,因此继续处理,将后继元素压入栈中。
数字 2 进栈,将栈顶元素与输出序列首元素比较,发现两者相等,因此将栈顶元素出栈,输出序列后移一位继续进行比较,随后重复上述过程,直到栈顶元素与输出序列的元素不相等,或者处理至栈空。那么继续处理,将后继元素压入栈中

数字 3 进栈;
数字 4 进栈;
数字 5 进栈,此时让栈顶的 5 出栈,继续让栈顶的 4 出栈,继续让栈顶的 3 出栈。

因此总的过程即是:
数字1进栈,数字2进栈,数字2出栈,数字1出栈,数字3进栈,数字4进栈,数字5进栈,数字5出栈,数字4出栈,数字3出栈。

程序结束,我们得到的出栈顺序为:2 1 5 4 3。

入栈出栈操作结束后,如果此时栈空,说明入栈的元素都一一按照指定顺序出栈了,则说明得到了给定的序列,输出“YES”

否则则输出“NO”;

完整代码实现:

#include<iostream>
#include<stack>
#include<algorithm>
using namespace std;
int num[105];
int main(){
    int N;
    while(scanf("%d",&N)==1&&N){
        stack <int> s;
        for(int i = 1;i <= N;i++){
            cin >> num[i];
        }
        int k = 1;
        for(int i = 1;i <= N;i++){
            s.push(i);
            while(!s.empty() && num[k]==s.top()){
                s.pop();
                k++;
            }
        }
        if(s.empty()){
            cout << "YES" << endl;
        }
        else{
            cout << "NO" << endl;
        }
    }
    return 0;
}

如有错误,还请指正,O(∩_∩)O谢谢

你可能感兴趣的:(栈,栈的简单应用,模拟入栈出栈过程)