题目描述:
某城市有一个火车站,有n节车厢从A方向驶入车站,按进站的顺序编号为1-n.你的任务是判断是否能让它们按照某种特定的顺序进入B方向的铁轨并驶入车站。例如,出栈顺序(5 4 1 2 3)是不可能的,但是(5 4 3 2 1)是可能的。
题目分析:
为了重组车厢,借助中转站,对于每个车厢,一旦从A移入C就不能回到A了,一旦从C移入B,就不能回到C了,意思就是A->C和C->B。而且在中转站C中,车厢符合后进先出的原则。故这里可以看做为一个栈。
解题思路:
当原数组元素等于目标数组元素时,先进栈再出栈即可,否则判断栈顶元素是否等于目标数组,若相等栈顶元素出栈,若不相等在数组不为空的情况下将原数组元素入栈。若都不满足说明不符合条件,flag = 0,break。
#include <iostream>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <stack>
using namespace std;
int target[101];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&target[i]);
stack <int> q;
int A=1,B=1; //A为原列车顺序 B为目标数组下表
bool ok=1;
while(B<n)
{
if(A==target[B]) {A++;B++;}
else if(!q.empty() && q.top()==target[B]) {q.pop();B++;}
else if(A<n) q.push(A++);
else {ok=0;break;}
}
printf("%s",ok ? "yes" : "no");
return 0;
}