HJ77 火车进站
描述
给定一个正整数N代表火车数量,0<N<10,接下来输入火车入站的序列,一共N辆火车,每辆火车以数字1-9编号,火车站只有一个方向进出,同时停靠在火车站的列车中,只有后进站的出站了,先进站的才能出站。
要求输出所有火车出站的方案,以字典序排序输出。
输入描述:
有多组测试用例,每一组第一行输入一个正整数N(0<n<10),第二行包括n个正整数,范围为1到9。< span="">
输出描述:
输出以字典序从小到大排序的火车出站序列号,每个编号以空格隔开,每个输出序列换行,具体见sample。
示例1
输入:
3 1 2 3
输出:
1 2 3 1 3 2 2 1 3 2 3 1 3 2 1
回溯算法,获得所有出栈序列
var fn = function(nums) {
let res = [];
let n = nums.length;
let backtrack = function(nums, stack, out){
//basecase
if(out.length === n){
res.push(out.join(' '));
return;
}
//选择1,可以入栈
if(nums.length){
stack.push(nums.shift());
backtrack(nums, stack, out);
nums.unshift(stack.pop())
};
//选择2,可以出栈
if(stack.length){
out.push(stack.pop())
backtrack(nums, stack, out);
stack.push(out.pop())
}
}
backtrack(nums, [], []);
return res.sort();
};
readline();
let nums = readline().split(' ')
fn(nums).forEach(arr=>console.log(arr))

浙公网安备 33010602011771号