题意:
输入数字n和k,n表示给出数组[1, 2, 3, 4 ... n],输出这个数组的第k个排列。
思路:
这种题目和“使用一些数字,找出它们能组成的第k小的数字”是一样的,很典型的数位dp问题。
假设我们已经确定了前i - 1位,现在确定第i位的数字是什么,也就是说现在要用剩下的n - i + 1个数字的第x个排列(x<k,因为前i -1位产生了一些已经比现在的排列小的排列,比如12345肯定比124xx小)。
那么我们可以将剩下的数字从大到小枚举,如果第i位填x,那么肯定比当前排列小的排列一共有多少个就可以求出(假设剩了数字2 4 6 8,那么如果填6,则肯定更小的排列数为 2 * 3!,即有2个比6小的数字,还剩3个数字全排列),这时选择最大的能使排列数<x的数字填在第i位上。
我在代码中用了presum表示前缀和,其实就是记录某个数字用没用过,也可以说记录某个数字之前还有几个数字没用过。
代码:
class Solution {
public:
string getPermutation(int n, int k) {
vector<int> presum;
vector<int> factorial;
for (int i = 0; i <= n; ++i) {
presum.push_back(i);
if (i) {
factorial.push_back(factorial[i - 1] * i);
} else {
factorial.push_back(1);
}
}
stringstream ss;
for (int i = 0; i < n; ++i) {
for (int j = n; j > 0; --j) {
if (presum[j] - presum[j - 1] && k > factorial[n - i - 1] * (presum[j] - 1)) {
ss << j;
k -= factorial[n - i - 1] * (presum[j] - 1);
for (int o = j; o <= n; ++o) {
--presum[o];
}
break;
}
}
}
return ss.str();
}
};