n個數全排列根據正數第m個排列找出倒數第m個排列(康託展開)
題目要求:把1~n的所有排列按字典序排成一排,從中選出一個排列,假設它是正數第m個排列,希望你能回答倒數的第m個排列是什麼?
例如1到3的所有排列是:
1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1
若選出的排列是1 2 3,則m=1,而你應該輸出的排列是3 2 1.
輸入描述:
第一行數字n,表示排列的長度
第二行n個數字,表示選出的排列,1<=n<=300000
輸出描述:
一行n個數字,表示所求的排列
題目分析思路:
- n個數的全排列,一共有n!(階乘)種可能的排列
- 根據康託展開的公式:X=an*(n-1)!+an-1*(n-2)!+…+ai*(i-1)!+…+a21!+a10!,反向計算某個排列所在的順序,
- 倒數順序=排列總數-正向順序+1,根據康託展開的正向思路,計算第幾個排列
全排列剖析:求n個數第k個排序----康託展開
Python編程代碼如下:
# encoding:utf-8
from functools import reduce
N = int(input())
A = [int(i) for i in input().split()]
def func(N):
# N個數排列組合總數
align_num = reduce(lambda x, y: x * y, range(1, N + 1))
# print("align_num:", align_num)
# 找到排列對應的是第幾個
index = find_align_index(N, A)
# print("index:",index)
res_arr = find_nth_align(N, align_num - index + 1)
print(" ".join(map(str, res_arr)))
return res_arr
# 計算階乘
def calc_factorial(n):
a = 1
for i in range(1, n + 1):
a = a * i
return a
# 查找某個排列對應的是第幾個
# 已知是n = 5,求14352是它的第幾個序列?
def find_align_index(n, arr):
index = 0
i = 1
while arr:
j = arr[0]
arr.remove(j)
count_min_j = len([i for i in arr if i < j])
index += count_min_j * calc_factorial(n - i)
i += 1
return index + 1
# 查找從n個數的排列中找第m個排列
def find_nth_align(n, m):
num_arr = list(range(1, n + 1))
m -= 1
arr = []
i = 1
while len(arr) < n - 1:
s = m // calc_factorial(n - i)
r = m % calc_factorial(n - i)
sort_arr = sorted(num_arr)
num = sort_arr[s]
arr.append(num)
num_arr.remove(num)
m = r
i += 1
arr.append(num_arr[0])
return arr
if __name__ == '__main__':
testcase1 = "2,2 1,1 2"
testcase2 = "3,1 2 3,3 2 1"
testcase3 = "4,1 3 2 4,4 2 3 1"
testcase4 = "5,3 1 5 2 4,3 5 1 4 2"
func(N)
運行結果: