算法分析与设计,第十一周博客。
60. Permutation Sequence
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):
"123"
"132"
"213"
"231"
"312"
"321"
Given n and k, return the kth permutation sequence.
Note: Given n will be between 1 and 9 inclusive.
要求我们给出关于n个数的第k个序列,其中序列是按照从小到大排列的。如果暴力破解的话,算法的时间复杂度是O(k),其中k是在[0, n!)之间摆动,当k越大时,算法的时间复杂度度就越趋近于O(n!),这样的算法的时间复杂度是很高的了。
那么就需要更好的方法,我们以n为3,即题目给出的例子为例,寻找下第k个序列的规律。
可以把这六个序列分成3部分,依据是序列的开头数字,分别是1、2、3,其中每个分组都有2个序列。那么,如果我们能够根据k的大小来找到它所属的序列的头个数字,那么就可以把这个范围缩小,进而减少了查找的范围。那么,如果根据k的大小确定首个数字呢?
其实也不难发现,对于n个数的全部序列,一共有n!个序列,而这些序列有可以分成n组,其中第i组的开头数字就是i,而这n个分组里所拥有的序列数目则是(n-1)!。而剩下的几个数字也可以用递归的思想用同样的方法找出来,这样这个问题就得到了解决。
总体的代码如下:
class Solution {
public:
string getPermutation(int n, int k) {
vector<int> fac(n+1);
vector<bool> used(n+1, false);
fac[0] = 1;
for (int i = 1; i <= n; ++i) {
fac[i] = fac[i-1]*i;
}
string res;
add(fac, used, n, k, res);
return res;
}
void add(const vector<int>& fac, vector<bool>& used, int n, int k, string& res) {
if (n == 0)
return ;
int one = (k-1)/fac[n-1];
int num = 0;
for (int i = 0, j = 0; i < used.size(); ++i) {
if (!used[i]) {
if (one == j)
{
num = i+1;
used[i] = true;
break;
}
++j;
}
}
res += to_string(num);
add(fac, used, n-1, k-one*fac[n-1], res);
}
};
我们来看下这个算法的时间复杂度。首先,是计算从0到n的阶乘,时间复杂度是O(n),然后是求得第k个序列,其中每调用一次
add函数,都会加入一个数字,所以一共调用了n次add函数,而add函数的时间复杂度呢,显然是O(n),所以寻找第k个序列的时间复
杂度是O(n^2),那么综合起来,这整个算法的时间复杂度是O(n^2)。相比于暴力解法的O(n!)来说是非常大的提升了。