排列问题（C++）

写在所有的前面：

本文采用C++实现代码

题目说明

题目

题目出处

广西大学oj
22级《算法设计与分析》第一次课堂作业
链接：https://oj.gxu.edu.cn/contest/619/problem/002

题目描述Description

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

输入Input

N<=10，K<=500000第一行为N和K,接下来的N行，每行N个数，0表示不能，1表示能

输出Output

N个数，空格分开
所求的排列

样例Sample

输入：

3 2
0 1 1
1 0 0
0 1 0

输出：

1 0 2

限制Hint

对样例的解释：
对于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

解答说明

方案1：逐位分配法（递归）

解题思路

保证待排序列由小到大
每一位从前到后取数（循环）
从前到后依次分配
满足条件就继续分配
分配完毕则计数器- -
计数器归零输出对应

一般情况

排列第a位
计数器k

判断是否排列到最后一位 if (a < n)
循环：从小到大选择一个数字作为当前位的值 for (int i = a; i < n; i++)
判断是否满足限制条件if (limit[list[a - 1]][list[i]] || a == 0)
如果满足：
把第i位放到第a位前面 exchange_in(a, i);
排列剩下的 prem(a + 1);
把第a位放回第i位后面 exchange_out(a, i);
排列完毕k --

特殊情况

计数器归零时退出递归 if (k == 0) return;

代码实现

#include<iostream>

using namespace std;

const int N = 100;

int limit[N][N];
int list[N];
int n, k;

//初始化 
void init()
{
	for (int i = 0; i < n; i++)
	{
		list[i] = i;
		for (int j = 0; j < n; j++)
			scanf("%d", &limit[i][j]);
	}
}

//打印数列
void print()
{
	for (int i = 0; i < n; i++)
		printf("%d ", list[i]);
	printf("\n");
}

//把i放到a前面 
void exchange_in(int a, int i)
{
	int tmp = list[i];
	for (int j = i; j > a; j--)
		list[j] = list[j - 1];
	list[a] = tmp;
}

//把a放到i后面
void exchange_out(int a, int i)
{
	int tmp = list[a];
	for (int j = a; j < i; j++)
		list[j] = list[j + 1];
	list[i] = tmp;
}

//全排列 
void prem(int a)
{
	if (k == 0) return;
	if (a < n)
	{
		for (int i = a; i < n; i++)
		{
			if (limit[list[a - 1]][list[i]] || a == 0)
			{
				exchange_in(a, i);
				prem(a + 1);
				exchange_out(a, i);
			}
		}
	}
	else 
	{
		k--;
		if (k == 0)
			print();
	}
}

int main()
{
	scanf("%d%d", &n, &k);

	init();

	prem(0);

	return 0;
}

其他解释