栈问题——铁轨

这个题代码我是直接抄的书上的，我只写写我的理解。

问题：

某城市有一个火车站，铁轨铺设如图6-1所示。有n节车厢从A方向驶入车站，按进站顺序编号为1～n。你的任务是让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。换句话说，在任意时刻，只有两种选择：A→C和C→B。

 

代码：

#include<stdio.h>
const int MAXN=1000+10;
int n,target[MAXN];
int main()
{
while(scanf("%d",&n)==1)
{
int stack[MAXN],top=0;
int A=1,B=1; //A为进入的车厢的编号
for(int i=1;i<=n;i++)
{
scanf("%d",&target[i]);          //输入车厢出站的顺序
}
int ok=1;
while(B<=n)
{
if(A==target[B])                      //判断进站车厢的编号与出站车厢的编号是不是相同
{
// printf("%d  %d  %d\n",A,B,target[B]);//调试语句

A++;B++;
}
else
    if(top&&stack[top]==target[B])//如果不同，判断栈顶与出站车厢编号是否相同
{
// printf("%d  %d  %d\n",top,stack[top],target[B]);//调试语句

top--; //出站（出栈）

B++; //下一个车厢进入
}
else
if(A<=n) //否则判断车辆是否全部进站
{
stack[++top]=A++;   //进站（进栈）
}
else
{
ok=0; //车厢全部进站后仍不符合
break;
}
}
printf("%s\n",ok?"Yes":"No");
}
}

STL代码

#include <cstdio>

#include <stack>

using namespace std;

const int MAXN = 1000 + 10;

 

int n, target[MAXN];

 

int main() {

  while(scanf("%d", &n) == 1) {

    stack<int> s;

    int A = 1, B = 1;

    for(int i = 1; i <= n; i++)

      scanf("%d", &target[i]);

    int ok = 1;

    while(B <= n) {

      if(A == target[B]){ A++; B++; }

      else if(!s.empty() && s.top() == target[B]){ s.pop(); B++; }

      else if(A <= n)  s.push(A++);

      else { ok = 0; break; }

    }

    printf("%s\n", ok ? "Yes" : "No");

  }

  return 0;

这两个代码一样，只是第二个包含了stack.h

只是加了一些注释，不过我相信这已足够，是这个程序更容易看懂了。