数据结构练习——车辆调度

---

一、题目描述

题目描述：

有一个火车站，铁路如图所示，每辆火车从A驶入，再从B方向驶出，同时它的车厢可以重新组合。假设从A方向驶来的火车有n节（n<=1000），分别按照顺序编号为1，2，3，…，n。假定在进入车站前，每节车厢之间都不是连着的，并且它们可以自行移动到B处的铁轨上。另外假定车站C可以停放任意多节车厢。但是一旦进入车站C，它就不能再回到A方向的铁轨上了，并且一旦当它进入B方向的铁轨，它就不能再回到车站C。 负责车厢调度的工作人员需要知道能否使它以a1,a2,…,an的顺序从B方向驶出，请来判断能否得到指定的车厢顺序。

输入：

输入文件的第一行为一个整数n，其中n<=1000，表示有n节车厢，第二行为n个数字，表示指定的车厢顺序。

输出:

如果可以得到指定的车厢顺序，则输出一个字符串”YES”，否则输出”NO”（注意要大写，不包含引号）。

样例输入：

6
1 5 3 2 4 6
1 2 3 4 5 6

样例输出：

YES

二、代码展示

注意注释，解题思路已体现在注释中

#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
#include<stdlib.h>
#include<stdbool.h>
#include<assert.h>
typedef int DataType;

typedef struct stack
{
	DataType* arr;
	int capacity;
	int top;
}ST;

//初始化栈
void StackInit(ST* st, size_t size)
{
	assert(st);
	st->arr = (DataType*)malloc(sizeof(DataType) * size);
	assert(st->arr);
	st->capacity = size;
	st->top = 0;
}
//入栈
void StackPush(ST* st, DataType x)
{
	assert(st);
	st->arr[(st->top)++] = x;
}
//出栈
void StackPop(ST* st)
{
	assert(st);
	st->top--;
}
//取栈顶元素
DataType StackTop(ST* st)
{
	assert(st);
	return st->arr[st->top-1];
}
//判断栈为空
bool IsEmpty(ST* st)
{
	assert(st);
	return st->top == 0;
}

int main()
{
	//读取所需数据
	int num = 0;
	scanf("%d", &num);
	int* input = (int*)malloc(sizeof(int) * num);
	for (int i = 0; i < num; i++)
	{
		scanf("%d", &input[i]);
	}
	int* output = (int*)malloc(sizeof(int) * num);
	for (int i = 0; i < num; i++)
	{
		scanf("%d", &output[i]);
	}
	//创建栈并初始化
	ST stack;
	StackInit(&stack, num);
	int i = 0, j = 0;
	while (i <= num && j <= num)
	{
		//如果输入和输出元素相同
		//应当先将输入元素入栈再出栈
		//相当于没有进行出入栈操作
		//直接将i和j后移即可
		if (input[i] == output[j])
		{
			i++;
			j++;
		}
		//如果输入和输出元素不同
		else
		{
			//如果栈不为空
			if (!IsEmpty(&stack))
			{
				//如果栈顶元素和输出元素相同
				//应当进行出栈操作，然后后移j
				//循环执行，直到栈顶元素和输出元素不同
				while (StackTop(&stack) == output[j])
				{
					StackPop(&stack);
					j++;
				}
			}
			//上述操作结束后，再将输入元素入栈并后移i
			StackPush(&stack, input[i]);
			i++;
		}
	}
	//判断栈是否为空
	//如果栈为空，则说明该输入输出匹配可以执行
	if (IsEmpty(&stack))
	{
		printf("YES\n");
	}
	//如果栈不为空，则说明该输入输出不匹配不可以执行
	else
	{
		printf("NO\n");
	}
	return 0;
}

---