给定 n 和 k,求123..n组成的排列中的第 k 个排列。
注意事项:
1 ≤ n ≤ 9
样例:
对于 n = 3, 所有的排列如下:
123
132
213
231
312
321
如果 k = 4, 第4个排列为,231.
挑战 :
O(n*k) in time complexity is easy, can you do it in O(n^2) or less?
思路:
不必将所有排列一一列出,由K值可以之间选出每一位的数字。
具体以n=4,k=10举例:
首先构造一个数组v,0,1,2,3,4,v[i]即可表示第i个数。在处理本位置时,计算出下一位置的排列数,n=4,(n-1)!=6,(k-1)/(n-1)!+1=2,故应选择v中第二个数,v[2]=2,此时结果为 “2” ;令k=k%(n-1)!=4,同时去掉v中第2个数,此时v,0,1,3,4;(n-2)!=2,(k-1)/(n-2)!+1=2,取v[2]=3,此时结果为 “23”,删除v[2],得v,0,1,4;k=k%(n-2)!=0,当k==0,说明是最后一个排列,则取v[v.size()-1],结果”234”;最终加上v[1],
得到最终结果”2341”。具体步骤见代码:
#ifndef C388_H
#define C388_H
#include<iostream>
#include<string>
#include<vector>
using namespace std;
class Solution {
public:
/*
* @param n: n
* @param k: the k th permutation
* @return: return the k-th permutation
*/
string getPermutation(int n, int k) {
// write your code here
string res;
if (n <= 0 || k <= 0)
return res;
vector<char> nums;//存放初始数组
int m = 1;
//m表示以nums中任一元素开头的排列数
for (int i = 1; i < n; ++i)
m *= i;
for (int i = 0; i <= n; ++i)
nums.push_back(i + '0');
//num为元素在nums中的位置,添加num对应的字符
//更新k与m的值
for (int i = 1; i < n; ++i)
{
int num = (k - 1) / m + 1;
if (k == 0)
num = nums.size() - 1;
res += nums[num];
k %= m;
m /= n - i;
nums.erase(nums.begin() + num);
}
res += nums[1];
return res;
}
};
#endif