SJTU火车栈

火车栈

有一条东西方向的铁路穿过小城A，小城A有一个火车调度站，示意图如下：

现在有N列火车自东向西依次开过来了，按照到达的先后次序编号为0号到N-1号。 根据调度局的要求，小城A的调度站要改变这些列车驶离A城的顺序。 为了达到这一目的， 调度站在任意时刻可以执行以下三种操作之一：

• 如果调度站还有剩余空间，则可以令下一列开来的火车进入调度站；
• 如果调度站内有列车，则可以令调度站最前方的火车离开调度站并驶离A城；
• 可以命令下一列开来的火车不经过调度站而直接驶离A城。

不过小城A的调度站实在太小了，只能容纳M列火车，请帮忙确认调度站能否完成任务。

例子
如果有4列火车开来，调度站可以容纳2列火车，调度局要求火车按照2、1、3、0的顺序驶离A城，则此任务可满足，一种可能的方案如下：

Step 1：火车0进入调度站；

Step 2：火车1进入调度站；

Step 3：火车2不经过调度站驶离A城；

Step 4：火车1从调度站驶离A城；

Step 5：火车3不经过调度站驶离A城；

Step 6：火车0从调度站驶离A城；

当然，你只需要回答是否可行，不需要列出一种可行方案。

Input Format
第一行是一个正整数T，表示本测试数据有多少个独立的测试点。（ T≤300）之后有T个独立的测试点，每个测试点占两行。 第一行有两个数字N和M，分别表示开来的火车数量，以及调度站最多可容纳的火车数量，两个数字之间用一个空格隔开。 第二行有N个整数，他们都在0到N−1之间，且不重复，用空格隔开，表示火车驶离A城的次序。N是正整数，且N≤1000；M是非负整数，且M≤1000。 M可能为0（这也许说明调度站的工作人员罢工了，或者正在这个考场考试）。

Output Format
输出共T行，每行对应一个测试点。如果能够调度，则回答YES，否则回答NO。 输出请注意大小写，每行行末直接回车，不要有其他字符。

Sample Input
2
4 2
2 1 3 0
5 2
2 4 3 1 0
Sample Output
YES
NO

#include <iostream>
#define elemType T
#define Top top_p
#define linkStack LinkStack
using namespace std;

template <class elemType>
class stack
{
public:
	virtual bool isEmpty() const = 0;
	virtual elemType  top() const = 0;
	virtual void push(const elemType &x) = 0;
	virtual elemType  pop() = 0;
	virtual ~stack() {}
};
template <class elemType>
class linkStack : public stack<elemType>
{
private:
	struct node {
		elemType  data;
		node *next;
		node(const elemType &x, node *N = NULL)
		{
			data = x; next = N;
		}
		node() :next(NULL) {}  //礼仪小姐专属，可以不要了
		~node() {}
	};

	node *Top;
public:
	linkStack() { top_p = NULL; };
	~linkStack() {
		node *tmp;
		while (Top != NULL) {//兄弟协同法
			tmp = Top;
			Top = Top->next;
			delete tmp;
		}
	}
	bool isEmpty()const;
	void push(const T&x);
	T pop();
	T top()const;
};

template<class T>
bool linkStack<T>::isEmpty()const
{
	return (top_p == NULL);
}

template<class T>
void linkStack<T>::push(const T&x)
{
	top_p = new node(x, top_p);
}

template<class T>
T linkStack<T>::pop()
{
	node*tmp = top_p;
	T x = top_p->data;
	top_p = top_p->next;
	delete tmp;
	return x;
}

template<class T>
T linkStack<T>::top()const
{
	return top_p->data;
}



struct node {
	int data;
	node*next;
	node(int d, node*p = NULL) { data = d; next = p; }
};

#include<iostream>
#include"linkStack.h"
int main() {
	int i, j, t, m, n, k, z,y[1000];//z计算栈内元素个数
	linkStack<int>x;
	cin >> t;
	for (i = 1; i <= t; i++)
	{
		z = 0;
		cin >> m >> n;
		int y[1000] = { 0 };
		for (j = 0; j <= m - 1; j++)
		{
			cin >> y[j];
		}
			for (k = 0; k <= m - 1; )
			{
				for (j = 0; j <= m - 1; j++)
				{
					while (z != 0 && x.top() == y[k]) { x.pop(); k++; z--; }
					if (j == y[k])
					{
						k++;
						continue;
					}
					if (j != y[k])
					{
						if (z == n) { break; }
						x.push(j); z++; continue;
					}
				}
				while (z != 0 && x.top() == y[k]) { x.pop(); k++; z--; }
				if (z == 0&&k==m) { cout << "YES" << endl; break; }
				else { cout << "NO" << endl; x.~LinkStack();  break; }
			}
		}
		system("pause");
	}