题目描述
有一个火车站,铁路如图所示,每辆火车从 A 驶入,再从 B 方向驶出,同时它的车厢可以重新组合。
假设从 A 方向驶来的火车有 n 节,分别按照顺序编号为 1,2,3,…,n。
假定在进入车站前,每节车厢之间都不是连着的,并且它们可以自行移动到 B 处的铁轨上。
另外假定车站 C 可以停放任意多节车厢。
但是一旦进入车站 C,它就不能再回到 A 方向的铁轨上了,并且一旦当它进入 B 方向的铁轨,它就不能再回到车站 C。
负责车厢调度的工作人员需要知道能否使它以 a1, a2, …, an 的顺序从 B 方向驶出,请来判断能否得到指定的车厢顺序。
输入格式
第一行为一个整数 n,表示有 n 节车厢,
第二行为 n 个数字,表示指定的车厢顺序。
输出格式
如果可以得到指定的车厢顺序,则输出一个字符串 YES
,否则输出 NO
输入样例
5
5 4 3 2 1
输出样例
YES
数据范围
n ≤ 1000
题解
stack:
解题思路
:
- 由于车厢按照编号依次入栈;
- 因此,若车厢
x
准备入栈,那么车厢1 ~ x-1
都已入栈,则不可能出现栈中元素 > 车厢x
的情况;
#include <iostream>
#include <stack>
using namespace std;
const int N = 1010;
int n;
int a[N];
int main()
{
cin >> n;
for (int i = 1; i <= n; i ++) cin >> a[i];
stack<int> stk;
for (int i = 1, k = 1; i <= n; i ++)
{
while(k <= a[i]) stk.push(k ++);
if(stk.top() == a[i]) stk.pop();
else
{
cout << "NO" << endl;
return 0;
}
}
cout << "YES" << endl;
return 0;
}