Tyvj P1336 火车进栈 (深搜 + 出栈序列判断)

题目链接:http://www.tyvj.cn/p/1336#


描述

有n列火车按1到n的顺序从东方左转进站,这个车站是南北方向的,它虽然无限长,
只可惜是一个死胡同,而且站台只有一条轨道,
火车只能倒着从西方出去,而且每列火车必须进站,先进后出。
(某生:不就是个栈吗?每次可以让右侧头火车进栈,或者让栈顶火车出站?
占卜哥:闭嘴!)
就像这样:
  出站<——-    <——进站
            |车|
            |站|
            |__|
现在请你按《字典序》输出前20种可能的出栈方案。
注意:这题当然不等于全排列!!!

输入格式

一个数n N<=20 数据保证不会TLE所以千万不要halt

输出格式

《字典序》输出前20种答案,每行一种,不要空格

测试样例1

输入

3

输出

123 
132 
213 
231 
321
1、先深搜全排列,然后判断全排列是否为合法出栈序列。简单题也要卡好久,唉。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
int n;
int num[25];
int vis[25];
int k;
int check(int * num) {
	int i, j;
	stack <int> s;
	for(i = 1, j = 1; i <= n;) {
		if(i == num[j]) {
			i++, j++;
		}
		else if(s.empty() || s.top() != num[j]){
			s.push(i);
			i++;
		}
		else if(s.top() == num[j]) {
			s.pop();
			j++;
		}
	}
	while(!s.empty()) {
		if(s.top() == num[j]) {
			s.pop();
			j++;
		}
		else return 0;
	}
	return 1;
}
void dfs(int pos) {
	int i;
	if(k == 20) return ;
	if(pos == n + 1) {
		if(check(num)) {
			k++;
			for(i = 1; i <= n; i++) {
				printf("%d", num[i]);
			}
			printf("\n");
		}
		return ;
	}
	for(i = 1; i <= n; i++) {
		if(vis[i] == 0) {
			vis[i] = 1;
			num[pos] = i;
			dfs(pos + 1);
			vis[i] = 0;
			num[pos] = 0;
		}
	}
}
int main() {
	scanf("%d", &n);
	k = 0;
	dfs(1);
	return 0;
}


你可能感兴趣的:(Tyvj P1336 火车进栈 (深搜 + 出栈序列判断))