Tsinghua MOOC 列车调度(Train)

描述

某列车调度站的铁道联接结构如图所示。

其中，A为入口，B为出口，S为中转盲端。所有铁道均为单轨单向式：列车行驶的方向只能是从A到S，再从S到B；另外，不允许超车。因为车厢可在S中驻留，所以它们从B端驶出的次序，可能与从A端驶入的次序不同。不过S的容量有限，同时驻留的车厢不得超过m节。

设某列车由编号依次为{1, 2, ..., n}的n节车厢组成。调度员希望知道，按照以上交通规则，这些车厢能否以{a1, a2, ..., an}的次序，重新排列后从B端驶出。如果可行，应该以怎样

的次序操作?

输入

共两行。

第一行为两个整数n，m。

第二行为以空格分隔的n个整数，保证为{1, 2, ..., n}的一个排列，表示待判断可行性的驶出序列{a1，a2，...，an}。

输出

若驶出序列可行，则输出操作序列，其中push表示车厢从A进入S，pop表示车厢从S进入B，每个操作占一行。

若不可行，则输出No。

输入样例1

5 2
1 2 3 5 4

输出样例1

push 
pop 
push 
pop
push
pop
push
push
pop
pop

输入样例2

5 5
3 1 2 4 5

输出样例2

No

限制

1 ≤ n ≤ 1,600,000

0 ≤ m ≤ 1,600,000

时间：2 sec

空间：256MB

思路

使用栈结构，模拟每一次列车的调度过程，如果中间模拟得不到对应的结果，则表明该序列无法实现；

具体来讲，对于样例2，第一个元素是3，如果要得到3，首先1,2,3压栈，然后即可弹出3，下一个需要得到1,然而栈顶元素是2，如果要得到1必须先弹出2,故该序列无法实现，输出No即可

代码

#include<stdio.h>
#include<stdlib.h>
#include<string.h>

int a[1600000];
char pint[3200000][5];
int n,m;
int pos_a=0,pos_s=0,pos_pint=0;

typedef struct{
	int *base;
	int *top;
	int stacksize;
}sqStack;
void initStack(sqStack *s)
{
	s->base = (int*)malloc((m+1)*sizeof(int));
	if(!s->base)exit(0);
	s->top = s->base;
	s->stacksize = m;
}
void Push(sqStack *s,int e)
{
	*(s->top) = e;
	s->top++;
	pos_s++;
	strcpy(pint[pos_pint++],"push");
}
void Pop(sqStack *s,int *e)
{
	if(s->top == s->base)return;
	*e = *--(s->top);
	pos_s--;
	strcpy(pint[pos_pint++],"pop");
}
int StackLen(sqStack s)
{
	return (s.top - s.base);
}
int main()
{
	int i=0,j=0;
	sqStack s;
	int temp_top;
	
	//freopen("in.txt","r",stdin);
	scanf("%d%d",&n,&m);
	while(i<n){scanf("%d",a+i);i++;}
	initStack(&s);
	for(i=1;pos_a<n;)
	{
		//printf("%d %d\n",i,a[pos_a]);
		if(i<a[pos_a])
		{
			if(m<pos_s+1)break;
			Push(&s,i);
			i++;
		}
		else if(i>a[pos_a])
		{
			Pop(&s,&temp_top);
			while(temp_top==a[pos_a]&&pos_s)
			{
				pos_a++;
				if(temp_top>a[pos_a])Pop(&s,&temp_top);
			}
			if(a[pos_a]<temp_top)break;
			if(i>n||!pos_s)pos_a++; 
		}
		else
		{
			if(m<pos_s+1)break;
			Push(&s,i);
			Pop(&s,&temp_top);
			pos_a++;i++;
		}
		
	}
	
	if(pos_a!=n)printf("No\n");
	else
	{
		i=0;
		while(i<pos_pint)
		puts(pint[i++]);
	}
	return 0;
}