铁轨

最新推荐文章于 2023-10-09 22:12:05 发布

原创最新推荐文章于 2023-10-09 22:12:05 发布 · 856 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

C/C++ 同时被 2 个专栏收录

194 篇文章

订阅专栏

Algorithm

38 篇文章

订阅专栏

本文介绍了一种利用栈结构解决车厢重组问题的算法实现方案，通过数组或STL中的stack来模拟中转站，确保车厢能够按照指定顺序出站。

某城市有一个火车站，有n节车厢从A方向驶入车站，按进站的顺序编号为1~n。现让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。在程序中输入车厢数目和出站的特定顺序，如果可以则输出Yes，否者输出No。
样例输入：
5
1 2 3 4 5
5
5 4 1 2 3
6
6 5 4 3 2 1
样例输出：
Yes
No
Yes

[分析]
因为栈只有一端生长 ，所以实现栈时只需要一个数组stack和栈顶指针top（始终指向栈顶元素） 。为了方便起见，这里使用的数组下标均从1开始。

[1] 使用数组模拟栈。

#include <cstdio> #define LOCAL const int maxn=1000+10; int n, target[maxn]; int main() { #ifdef LOCAL freopen("output.txt","a+",stdout); #endif while (scanf_s("%d",&n)==1) { int stack[maxn], top=0, 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])// ok { ++A; ++B; } else if (top && stack[top]==target[B])// pop stack { --top; ++B; } else if (A<=n)// push stack { stack[++top]=A++; } else// No { ok=0; break; } } printf("%s/n",ok?"Yes":"No"); } return 0; }

[2] 使用STL中的stack实现。
#include <cstdio> #include <stack> using std::stack; 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])// ok { ++A; ++B; } else if (!s.empty() && s.top()==target[B])// pop stack { s.pop(); ++B; } else if (A<=n)// push stack { s.push(A++); } else// No { ok=0; break; } } printf("%s/n",ok?"Yes":"No"); } return 0; }