这道题可以使用回溯的方法解决,为了避免超时,可以将回溯转化成递归。
因为结果是由1~n这n个数来组成的,并且没有重复,所以我们可以一位一位地来确定。n个数的全排列是 n! 个,由于是有序的,那么如果确定了第一位,后面可能的结果就变成 (n-1)! 个,范围就变小了。
下面通过一个例子来说明这个问题:
如果 n = 3, k = 3
所有的排列是
1 2 3 1 3 2 2 1 3 2 3 1 3 1 2 3 2 1
通过算数运算可以发现 k = 3这个位置处于第二个区域里面,也就意味着结果的首元素为2,确定了结果的首元素,下一步就在第二个区域里面细分,找出第二位是什么,以此类推,最后得出答案。
import java.util.ArrayList; import java.util.List; class Solution { public String getPermutation(int n, int k) { StringBuilder res = new StringBuilder(); List<Integer> num = new ArrayList<>(); for(int i = 1; i <= n; i++){ num.add(i); } //num = [1,2,3,...,n] int[] factorial = new int[n]; factorial[0] = 1; for(int i = 1; i < n; i++){ factorial[i] = i * factorial[i - 1]; } n--; while(n >= 0){ double t = factorial[n]; int loc = (int)(Math.ceil(k / t)) - 1; if(-1 == loc){ loc = num.size() - 1; } res.append(num.get(loc)); num.remove(loc); k %= t; n--; } return res.toString(); } }