31. Next Permutation + 46. Permutations + 47. Permutations II + 60. Permutation Sequence

▶ 问题：字典序生成有关的问题。

▶ 31. 由当前序列生成字典序里的下一个序列。

● 初版代码，19 ms

class Solution
{
public:
    void nextPermutation(vector<int>& nums)
    {
        int i, j, temp;
        for (i = nums.size() - 1; i > 0 && nums[i - 1] >= nums[i]; i--);// 从右向左寻找第一个递增对
        if (!i)                                     // i == 0，当前排序已经是最大的，全反序得到最小的
        {
            for (j = 0; i < nums.size() / 2; j++)
                temp = nums[j], nums[j] = nums[nums.size() - 1 - j], nums[nums.size() - 1 - j] = temp;
            return;
        }
        else if (i == nums.size() - 1)              // 最后两个元素就是递增对，交换他俩就行
        {
            temp = nums[i], nums[i] = nums[i - 1], nums[i - 1] = temp;
            return;
        }        
        for (j = nums.size() - 1; j >= i && nums[j] <= nums[i - 1]; j--);// 其他情形，需要在右侧递减部分中找到比递增较小元更大的第一个元素
        temp = nums[i - 1], nums[i - 1] = nums[j], nums[j] = temp;       // 将右侧较大的元素交换到递增对较小元的位置
        for (j = i; j < (nums.size() + i) / 2; j++)                      // 右侧进行反序
            temp = nums[j],nums[j] = nums[nums.size() + i - 1 - j],nums[nums.size() + i - 1 - j] = temp;
        return;
    }
};

● 改进代码，15 ms，第二次查找时使用二分法改进，引入原数组长度的常量，引入内置交换函数 std::swap()。

class Solution
{
public:
    void nextPermutation(vector<int>& nums)
    {
        const int len = nums.size();
        int i, j, temp, lp;
        for (i = len - 1; i > 0 && nums[i - 1] >= nums[i]; i--);
        if (!i)
        {
            for (i = 0; i < len / 2; i++)
                swap(nums[i], nums[len - 1 - i]);
            return;
        }
        else if (i == len - 1)
        {
            swap(nums[i], nums[i - 1]);
            return;
        }
        for (lp = i, j = len, temp = (lp + j) / 2; temp > lp; temp = (lp + j) / 2)
            (nums[temp] > nums[i - 1]) ? (lp = temp) : (j = temp);
        swap(nums[i - 1], nums[lp]);
        for (j = i; j < (len + i) / 2; j++)
            swap(nums[j], nums[len + i - 1 - j]);
        return;
    }
};

● 大佬的代码，13 ms，使用了排序函数

class Solution
{
public:
    void nextPermutation(vector<int>& nums)
    {
        const int len = nums.size();
        int i, flag, index;
        for (flag = 1, i = index = len - 1; i > 0; i--)
        {
            if (nums[i] <= nums[i - 1])
                index--;
            else
            {
                for (int j = len - 1; j >= index; j--)
                {
                    if (nums[j] > nums[i - 1])
                    {
                        swap(nums[i - 1], nums[j]);
                        break;
                    }
                }
                sort(nums.begin() + i, nums.end());
                flag = 0;
                break;
            }
        }
        if (len && flag)
            sort(nums.begin(), nums.end());
    }
};

 ● 偷懒的代码，14 ms，直接使用内置函数 next_permutation()，该函数返回值是一个 bool 变量，仅当输入的数组引用是一个降序时返回 false，但此时仍然将数组调整为字典序的下一个（升序）。

class Solution
{
public:
    void nextPermutation(vector<int>& nums)
    {
        next_permutation(nums.begin(), nums.end());
        return;
    }
};

 

▶ 46. 生成给定元素的所有排列（元素无重复）

● 自己的代码，13 ms，反复调用第 31 题的结果来进行枚举。

class Solution
{
public:
    void nextPermutation(vector<int>& nums)
    {
        const int len = nums.size();
        int i, j, temp, lp;
        for (i = len - 1; i > 0 && nums[i - 1] >= nums[i]; i--);
        if (!i)
        {
            for (i = 0; i < len / 2; i++)
                swap(nums[i], nums[len - 1 - i]);
            return;
        }
        else if (i == len - 1)
        {
            swap(nums[i], nums[i - 1]);
            return;
        }
        for (lp = i, j = len, temp = (lp + j) / 2; temp > lp; temp = (lp + j) / 2)
            (nums[temp] > nums[i - 1]) ? (lp = temp) : (j = temp);
        swap(nums[i - 1], nums[lp]);
        for (j = i; j < (len + i) / 2; j++)
            swap(nums[j], nums[len + i - 1 - j]);
        return;
    }
    vector<vector<int>> permute(vector<int>& nums)
    {
        int count, i;
        vector<vector<int>> output;
        for (i = nums.size(), count = 1; i > 1; count *= i, i--);   // 计算阶乘（总共排列数）
        sort(nums.begin(),nums.end());                              // 排序得到第一个排列
        output.push_back(nums);
        for (i = 1; i < count; i++)                                 // 每次循环生成下一个排列，不合并压入可以优化最后一次函数 nextPermutation()调用
        {
            nextPermutation(nums);
            output.push_back(nums);
        }          
        return output;
    }
};

● 大佬的代码，13 ms，使用了内置函数 next_permutation()

class Solution
{
public:
    vector<vector<int>> permute(vector<int>& nums)
    {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        for (result.push_back(nums); next_permutation(nums.begin(), nums.end());)
            result.push_back(nums);
        return result;
    }
};

 

▶ 47. 生成给定元素的所有排列（元素有重复）

● 自己的代码，26 ms，原数组排序后统计各重复部分的长度，计算重复排列数。由于前面 31 题中的程序在求“下一个排列”的时候已经跳过了重复的情形，所以只要恰当的减少该函数的调用次数就能列出所有含重复元素的排列。

class Solution
{
public:
    void nextPermutation(vector<int>& nums)// 改进
    {
        const int len = nums.size();
        int i, j, temp, lp;
        for (i = len - 1; i > 0 && nums[i - 1] >= nums[i]; i--);
        if (!i)
        {
            for (i = 0; i < len / 2; i++)
                swap(nums[i], nums[len - 1 - i]);
            return;
        }
        else if (i == len - 1)
        {
            swap(nums[i], nums[i - 1]);
            return;
        }
        for (lp = i, j = len, temp = (lp + j) / 2; temp > lp; temp = (lp + j) / 2)
            (nums[temp] > nums[i - 1]) ? (lp = temp) : (j = temp);
        swap(nums[i - 1], nums[lp]);
        for (j = i; j < (len + i) / 2; j++)
            swap(nums[j], nums[len + i - 1 - j]);
        return;
    }
    inline int factorial(int a)// 阶乘函数
    {
        int i, output;
        for (output = 1, i = 2; i <= a; output *= i, i++);
        return output;
    }
    vector<vector<int>> permuteUnique(vector<int>& nums)
    {
        int count, i, rep;
        vector<vector<int>> output;
        count = factorial(nums.size());                         // 计算重复排列数
        sort(nums.begin(), nums.end());                         // 排序得到第一个排列
        for (i = 0; i < nums.size() - 1; i++)
        {
            for (; i < nums.size() - 1 && nums[i] != nums[i + 1]; i++);               // 跳过相邻项不同的部分
            for (rep = 0; i < nums.size() - 1 && nums[i] == nums[i + 1]; rep++, i++); // 记录本节中相邻项相同的个数            
            count /= factorial(rep + 1);                                              // 重复排列的计算
        }
        output.push_back(nums);
        for (i = 1; i < count; i++)                            // 每次循环生成下一个排列，不合并压入可以优化最后一次函数 nextPermutation()调用
        {
            nextPermutation(nums);
            output.push_back(nums);
        }
        return output;
    }
};

 ● 大佬的代码，30 ms，回溯，逐个寻找新排列。

class Solution
{
public:
    vector<vector<int>> permuteUnique(vector<int>& nums)
    {
        vector<vector<int>> result;                // 存放结果
        vector<int> flags(nums.size(), 0), content;// flag[i] 表当前 nums[i] 是否被使用，content 为当前排列（填满就送入 reslut）
        sort(nums.begin(), nums.end());
        helper(nums, content, flags, result);
        return result;
    }

    void helper(vector<int>& nums, vector<int>& content, vector<int>& flags, vector<vector<int>>& result)
    {
        if (content.size() == nums.size())// 所有元素都被使用，将 content 中的排列放入 result 中
        {
            result.push_back(content);
            return;
        }
        for (int i = 0; i < nums.size(); i++)
        {
            if (i > 0 && nums[i] == nums[i - 1] && flags[i - 1] != 0)// 跳过相同的元素
                break; 
            if (!flags[i])// 元素 nums[i] 还没有被谁用过
            {
                flags[i] = 1;              // 把元素 nums[i] 添加到当前的排列末尾
                content.push_back(nums[i]);
                int i1 = i;
                while (i < nums.size() - 1 && nums[i + 1] == nums[i])// 跳过相同元素
                    i++;
                helper(nums, content, flags, result); // 尝试在添加了元素 nums[i] 以后继续添加下一个元素
                content.pop_back();                   // 去掉当前添加的元素，由于元素可能重复，所以需要在最后一个元素处标记未使用
                flags[i1] = 0;
            }
        }
    }
};

● 大佬的代码改进，27 ms，基本算法相同。

class Solution
{
public:
    vector<vector<int>> permuteUnique(vector<int>& nums)
    {
        vector<vector<int>> result;               
        vector<bool> flags(nums.size(), true);
        vector<int> content;
        sort(nums.begin(), nums.end());
        helper(nums, content, flags, result);
        return result;
    }
    void helper(vector<int>& nums, vector<int>& content, vector<bool>& flags, vector<vector<int>>& result)
    {
        if (content.size() == nums.size())
        {
            result.push_back(content);
            return;
        }
        int i, j;
        for (i = 0; i < nums.size(); i++)
        {
            if (flags[i])
            {
                flags[i] = false;
                content.push_back(nums[i]);                
                for (j = i; i < nums.size() - 1 && nums[i + 1] == nums[i]; i++);
                helper(nums, content, flags, result);
                content.pop_back();
                flags[j] = true;
            }
        }
    }
};

 

▶ 60. 求给定元素字典序排列中的第 k 个。

● 初版，108 ms，逐个生成排列，直到第 k 个为止。

class Solution
{
public:
    string getPermutation(int n, int k)
    {
        int i;
        vector<int> table(n);
        for (i = 0; i < n; table[i] = i++);
        for (i = 1; i < k; i++)
            next_permutation(table.begin(), table.end());
        string output("");
        for (i = 0; i < n; i++)
            output += table[i] + '1';
        return output;
    }
};

● 改进版，6 ms，利用阶乘计算出所求排列，最快的解法算法与之相同。

class Solution
{
public:
    string getPermutation(int n, int k)
    {
        int i, j;
        string ret;
        vector<int> factorial(n, 1);// 阶乘和字符的打表
        vector<char> num(n, 1);

        for (i = 1; i < n; i++)
            factorial[i] = factorial[i - 1] * i; 
        for (i = 0; i < n; i++)
            num[i] = '0' + (i + 1);
        for (i = n, k--; i >= 1; i--)
        {
            j = k / factorial[i - 1];
            k %= factorial[i - 1];
            ret.push_back(num[j]);
            num.erase(num.begin() + j);
        }
        return ret;
    }
};