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