LeetCode 60 Permutation Sequence

来源:互联网 发布:app推广 aso优化方案 编辑:程序博客网 时间:2024/05/10 20:19

题意:

输入数字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();    }};




原创粉丝点击