蓝桥杯---排列问题---全排列

试题 算法训练 排列问题

资源限制

时间限制：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

实现代码

#include<iostream>
#include<algorithm>
using namespace std;

int n, k, cnt = 0;
int arr[10], vis[10][10];

bool Judge() {
	for(int i = 1; i< n; i++) if (!vis[arr[i - 1]][arr[i]]) return false;
	return true;
}

int main() { 
	cin >> n >> k;
	for (int i = 0; i < n; i++) arr[i] = i;
	for (int i = 0; i < n; i++) for (int j = 0; j < n; j++) cin >> vis[i][j];
	do {
		if (Judge()) if (++cnt == k) break;
	} while (next_permutation(arr, arr + n));
	for (int i = 0; i < n; i++) {
		if (i) cout << " ";
		cout << arr[i];
	}
	cout << endl;
	return 0;
}