算法训练 排列问题

题目链接

问题描述

　　求一个0～N-1的排列（即每个数只能出现一次），给出限制条件（一张N*N的表，第i行第j列的1或0，表示为j-1这个数不能出现在i-1这个数后面，并保证第i行第i列为0），将这个排列看成一个自然数，求从小到大排序第K个排列。

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 10+5;
int nexts[N][N], t[N];
int n, k, cnt = 0;

int main(int argc, char** argv) {	
	cin>> n>> k;	
	for(int i = 0; i < n; i++){
		t[i] = i;
		for(int j = 0; j < n; j++)
			cin>> nexts[i][j];
	}
	do{
		bool ok = true;
		for(int i = 1; i < n; i++){
			if(nexts[t[i-1]][t[i]] == 0){
				ok = false;
				break;
			}
		}
		if(ok) cnt++;
		if(cnt == k)  break;
	}while(next_permutation(t, t+n));
	for(int i = 0; i < n; i++)
		cout<< t[i]<< " ";			 
	return 0;
}

 

数据规模和约定

　　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