题目描述:
输入格式:
输入有两行,第一行是一个正整数 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; }