Java实现 蓝桥杯VIP 算法训练 排列问题

算法训练 排列问题
时间限制：1.0s 内存限制：512.0MB
问题描述
　　求一个0～N-1的排列（即每个数只能出现一次），给出限制条件（一张N*N的表，第i行第j列的1或0，表示为j-1这个数不能出现在i-1这个数后面，并保证第i行第i列为0），将这个排列看成一个自然数，求从小到大排序第K个排列。
数据规模和约定
　　N<=10，K<=500000
输入格式
　　第一行为N和K,接下来的N行，每行N个数，0表示不能，1表示能
输出格式
　　所求的排列
样例输入
3 2
0 1 1
1 0 0
0 1 0

样例输出

1 0 2
解释：
对于N=3的没有任何限制的情况
第一：0 1 2
第二：0 2 1
第三：1 0 2
第四：1 2 0
第五：2 0 1
第六：2 1 0
根据题目所给的限制条件由于2不能出现在1后面，0不能出现在2后面
第一：0 2 1
第二：1 0 2
第三：2 1 0

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
import java.util.Collections;
import java.util.LinkedList;


public class 排列问题 {

	private static int vis[][];
	private static int count = 0, m;
	private static LinkedList<String> queue = new LinkedList<String>();

	public static void main(String[] args) throws IOException {

		StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));

		st.nextToken();
		int n = (int) st.nval;
		st.nextToken();
		m = (int) st.nval;

		int arr[] = new int[n];
		vis = new int[n][n];
		for (int i = 0; i < n; i++) {
			arr[i] = i;
			for (int j = 0; j < n; j++) {
				st.nextToken();
				vis[i][j] = (int) st.nval;
			}
		}
		allsort(arr, 0);
		Collections.sort(queue);
		System.out.println(queue.get(m - 1));
	}

	private static void allsort(int arr[], int k) {

		if (arr.length == k) {
			int ok = 0;
			String str = "";
			for (int i = 0; i < arr.length - 1; i++) {
				if (vis[arr[i]][arr[i + 1]] == 0) {
					ok = 1;
					return;
				}
				str += arr[i] + " ";
			}

			if (ok == 0) {
				queue.add(str + arr[k - 1]);
			}
			return;
		}

		int tmp;
		for (int i = k; i < arr.length; i++) {
			tmp = arr[i];
			arr[i] = arr[k];
			arr[k] = tmp;
			allsort(arr, k + 1);
			tmp = arr[i];
			arr[i] = arr[k];
			arr[k] = tmp;
		}
	}

}