【LeetCode】60. 第k个排列 (C++)

原题地址：https://leetcode-cn.com/problems/permutation-sequence/

题目描述：

给出集合 [1,2,3,…,n]，其所有元素共有 n! 种排列。

按大小顺序列出所有排列情况，并一一标记，当 n = 3 时, 所有排列如下：

    "123"
    "132"
    "213"
    "231"
    "312"
    "321"

给定 n 和 k，返回第 k 个排列。

说明：

    给定 n 的范围是 [1, 9]。
    给定 k 的范围是[1,  n!]。

示例 1:

输入: n = 3, k = 3
输出: "213"

示例 2:

输入: n = 4, k = 9
输出: "2314"

解题方案：

刚开始用了全排列，虽然结果正确但是超时了，然后就考虑用数学规律直接推算。要注意的一点是恰好能整除的情况，我开始时就没有写这个判断。

代码：

class Solution {
public:
    string getPermutation(int n, int k) {
        vector<int> allFac(1, 1);
        for(int i = 1; i <= n; i ++)
            allFac.push_back(factorial(i));
        vector<int> num;
        for(int i = 1; i <= n; i ++)
            num.push_back(i);
        int sum = 0;
        for(int i = 1; i <= n; i ++)
        {
            int a = k / allFac[n - i];
            int b = k % allFac[n - i];
            int pos;
            if(b != 0)
                pos = a;
            else
                pos = a - 1;
            sum = sum * 10 + num[pos];
            cout << num[pos];
            k -= pos * allFac[n - i];
            num.erase(num.begin() + pos);
            
        }
        
        string s;
        stringstream ss;
        ss << sum;
        ss >> s;
        return s;
    }
    int factorial(int n)
    {
        int sum = 1;
        for(int i = 1; i <= n; i ++)
            sum *= i;
        return sum;
    }
};