试题 算法训练 排列问题

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n,k,x;
	int cnt=0,s[109][3]={0};
	scanf("%d %d",&n,&k);
	for(int i=0;i<n;i++)
	 for(int j=0;j<n;j++){
	 	scanf("%d",&x);
	 	if(i==j||x==1) continue;
	 	s[cnt][0]=j;
	 	s[cnt++][1]=i;
	 }
	int b[10],num=0;
	for(int i=0;i<n;i++)
	 b[i]=i;
	do{
		int l1=0,l2=0,flag=0;
		for(int i=0;i<cnt;i++){
			for(int j=0;j<n;j++){
				if(b[j]==s[i][0]) l1=j;
				if(b[j]==s[i][1]) l2=j; 
			}
			if(l1-1==l2)
			 flag=1;
			if(flag==1)
			 break;
		}
		if(flag==0)
		 num++;
		if(num==k){
			for(int i=0;i<n;i++){
				if(i) cout<<" ";
				cout<<b[i];
			}
			break;
		}
	}while(next_permutation(b,b+n));
	return 0;
}