一.问题描述
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.
二.我的解题思路
最直观的办法就是挨个生成
permutation,然后生成到第k个的时候输出即可,之前好像做过一个类似的叫做next permutation的问题,利用这样的思路应该也可以解决本问题。
另外一种思路就是找本题的规律。以[1,2,3]为例,上面问题描述中已经给出了具体的6种顺序组合。那么这样的6组数据有什么规律呢?首先看“123”和“132”,可以看出他们是“1 ”+[2,3]的顺序排列,同样的,“213”和“231”,可以看出他们是“2 ”+[1,3]的顺序排列。这样就依稀看到了递归的影子,原问题和子问题只有规模不同,别的都相同,那么我们就可以用分治递归法进行求解。最后的程序如下:
class Solution {
public:
string getPermutation( int n, int k) {
string res; vector<int> data;
for (int i = 0; i<n; i++) data.push_back(i + 1);
int * record = new int (n + 1);
int tmp = 1;
for (int i = 1; i <= n; i++){
tmp = tmp*i;
record[i] = tmp;
}
if(k>record[n]) return res;
gen_res(data, k, record, res);
return res;
}
void gen_res(vector<int> &data, int k, int * record, string &res){
int dist = data.size() - 1; int new_k;
if (dist == 0) {
char curr = '0' + data[0];
res += curr;
return;
}
int yushu = k % record[dist] == 0 ? -1 : 0;
int idx = k / record[dist] + yushu;
char curr = '0' + data[idx];
vector<int>::iterator it = data.begin();
data.erase(it + idx);
res += curr;
if (yushu == -1) yushu = record[dist];
else yushu = k % record[dist];
gen_res(data, yushu, record, res);
}
};
这里record数组负责记录n的阶乘,data负责记录当前问题中余下的数字。上述程序自己测试没有问题,但是在leetcode oj时候会报runtime error,暂时还不知道为何。。