算法竞赛入门经典 6.1.2 （Y型）铁轨

最新推荐文章于 2021-03-02 20:37:35 发布

原创最新推荐文章于 2021-03-02 20:37:35 发布 · 647 阅读

0 ·

CC 4.0 BY-SA版权

竞赛入门 acm 专栏收录该内容

31 篇文章

订阅专栏

本文通过三种不同的方式实现了列车调度算法：使用基于数组的栈、简洁版基于数组的栈以及利用STL中的stack。每种实现都展示了如何判断是否可以通过一系列的入栈和出栈操作使得列车按指定顺序离开车站。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//自己写的，用array-based stack来解题代码（逻辑不够简洁清晰）
#include <stdio.h>
const int MAXN = 100;
int stack[MAXN];
int target[MAXN];
int main()
{
	int n = 0;
	while(scanf("%d", &n) != EOF)
	{
		for(int i = 0; i < n; ++i)
			scanf("%d", &target[i]);
		int top = 0;
		int cur = 1;
		int k = 0;
		while(cur == target[k] || (top && stack[top] == target[k]) ||  cur <= n)
		{
			if(cur == target[k])
			{
				++cur;
				++k;
			}
			else if(top && stack[top] == target[k])
			{
				--top;
				++k;
			}
			else
				stack[++top] = cur++;
		}
		if(!top && cur > n)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}

//书本上，用array-based stack来解题（简洁）
#include <stdio.h>
const int MAXN = 100;
int n = 0, stack[MAXN];
int main()
{
	while(scanf("%d", &n) != EOF)
	{
		int top = 0, target[MAXN];
		for(int i = 0; i < n; ++i)
			scanf("%d", &target[i]);
		int cur = 1;//现在几号车进站
		int k = 0;//读到第几个应该出站的车的车号
		bool ok = 1;
		while(k < n)
		{
			if(cur == target[k])
			{
				++cur;
				++k;
			}
			else if(top && stack[top] == target[k])
			{
				--top;
				++k;
			}
			else if(cur <= n)
				stack[++top] = cur++;
			else
			{
				ok = 0;
				break;
			}
		}
		printf("%s\n", ok ? "Yes" : "No");
	}
	return 0;
}

//使用STL的stack解题
#include <cstdio>
#include <stack>
using namespace std;
const int MAXN = 100;
int n = 0, target[MAXN];
int main()
{
	while(scanf("%d", &n) != EOF)
	{
		for(int i = 0; i < n; ++i)
			scanf("%d", &target[i]);
		stack<int> s;
		int cur = 1;
		int k = 0;
		bool ok = true;
		while(k < n)
		{
			if(cur == target[k])
			{
				++k;
				++cur;
			}
			else if(!s.empty() && s.top() == target[k])
			{
				++k;
				s.pop();
			}
			else if(cur <= n)
				s.push(cur++);
			else
			{
				ok = false;
				break;
			}
		}
		printf("%s\n", ok ? "Yes" : "No");
	}
	return 0;
}