The set [1,2,3,…,n]
contains a total of n! unique permutations.
By listing and labeling all of the permutations in order,
We get the following sequence (ie, for n = 3):
1 123 2 132 3 213 4 231 5 312 6 321
Given n and k, return the kth permutation sequence.
Note: Given n will be between 1 and 9 inclusive.
题解:
开始想用暴力来着,分别用了STL中的next_permutation函数和自己写的字典序实现的next_permutation函数,结果都TLE在n=9这里了,最后上网找了资料才发现这题是有规律的。
以n=3为例,我们用一个数组nums放1,2,3,所以nums = [1,2,3],假设k=4
那么如果先不考虑第一位,后n-1位总共的排列有(n-1)!个,所以第一位数就可以算出来在nums中的index就是(k-1)/(n-1)!,本例中就是(4-1)/2!=1,所以第一位数是nums[1]=2。这里k为什么要减1呢?因为如果k不减1,那么k分别取1,2,3,4,5,6时,我们得到的第一位数分别在nums中的位置是0,1,1,2,2,3,发现正好“错位”,实际应该是0,0,1,1,2,2,把k-1就正好可以得到正确的解。
然后看第二位,注意在算第二位数字前要先更新k和nums,更新k’=(k-1)/(n-1)!=3/2 =1;更新nums,把2后面的每一个数往前移动一位,覆盖前面的数,本例得到的新的nums=[1,3],表示剩下的几位数在1,3中取。那么第二位数在nums中的index就可以算出来是k’/(n-2)!=1/1=1,即nums[1]=3。
最后一位数就是nums中剩下的1了。
代码如下:
1 class Solution { 2 public: 3 string getPermutation(int n, int k) { 4 int f = 1; 5 string answer = ""; 6 7 for(int i = 1;i<=n-1;i++){ 8 f *= i; 9 } 10 //f = (n-1)! 11 k--; 12 13 int nums[10] = {1,2,3,4,5,6,7,8,9}; 14 for(int i = 0;i < n;i ++){ 15 answer += nums[k/f] + '0'; 16 17 for(int j = k/f;j < n;j ++) 18 nums[j] = nums[j+1]; 19 20 k = k%f; 21 if(n-i-1 > 0) 22 f = f/(n-i-1); 23 } 24 return answer; 25 } 26 };
代码解释:
第7行循环计算(n-1)!;
15行计算第i位上的数;
17行循环更新nums数组;
20~23行更新k和f,f存放各阶的阶乘,在循环的过程中从(n-1)!一直到1;
注意21行有个判断n-i-1是否大于0,因为最后一次循环,i=n-1,如果不判断,会出现除以0操作,程序会奔溃。