问题 G: 车站调度

前言必读！http://blog.csdn.net/hnust_v/article/details/51747743

问题 G: 车站调度

题目描述

有顺序排列的1,2, 3,…,n节车厢在入站口等待调度。车站设置了一个栈作为缓冲，这样的话只可能进行下列两个操作之一：

（1）如果还有车厢在入站口，将最前面的入栈缓冲

（2）将栈顶的车厢驶出车站

给定一个1至n的排列，问其作为出站序列是否合法。

注意：入站顺序为1,2, 3,…,n，即1先入栈…，n最后入栈。

输入

输入包含若干测试用例。每一个测试用例由多行组成。第一行是两个整数n（1<=n <= 100）和m，n表示入站序列为1至n。m表示随后有m行出站序列。

当n，m均为0时表示输入结束。

输出

对应每一个出站序列，合法则输出一行YES，否则输出一行NO。

//思路：反题意考虑所给序列能不能通过车站（栈）还原成 n,n-1,n-2,……1序列
令i=n
第一步 应当让i先出站
判断 i是否为0
是       返回正确
不是   执行第二步
第二步 判断当前最后一辆是不是i?
是
i=i-1
执行第三步
第三步 判断栈顶元素是不是i
是
i出栈
i=i-1
执行第四步
不是  执行第四步
第四步 判断当前车站（栈）是否为空
是
将i压入栈内
i=i-1并且返回第一步
不是
判断栈顶元素是否比当前最后一辆序列大
是       返回错误 //因为若序号低的压在序号高的上面就无法先输出序号高的，必然是错误的
不是   将最后一辆序号压入栈内
i=i-1并且返回第一步

伪代码描述 :

Judge (n, stack<int> S)： //stack请按书实现功能 或者使用 C++ STL
need=n
for i=n to 1
if A[i] = need
need = need -1
while !S.empty() and S.top = need
S.pop()
need = need -1
continue
if S.empty()
S.push(A[i])
else
if S.top() > A[i]
return false
else S.push(A[i])
return true

//C++:
#include <bits/stdc++.h>
using namespace std;
int A[3000];
int Judge(int n)
{
stack<int> S;
int now=n;

for(int i=n;i>=1;i--)
{
if(A[i]==now)
{
now--;
while(!S.empty()&&S.top()==now)
{
S.pop();
now--;
}
continue;
}

// if(!S.empty()) cout<<S.top()<<endl;
if(S.empty()) S.push(A[i]);
else
{
if(S.top()>A[i]) return 0;
else S.push(A[i]);
}
}
return 1;
}
int main()
{
//freopen("F:\\test.txt","r",stdin);
int n,m;
while(~scanf("%d %d",&n,&m)&&(n||m))
{
for(int k=1;k<=m;k++)
{
for(int i=1;i<=n;i++) scanf("%d",&A[i]);
// printf("\nCase %d\n",k);
printf("%s\n",Judge(n)?"YES":"NO");
}
}
}