木又同学2020年第17篇解题报告
leetcode第31题:下一个排列
https://leetcode-cn.com/problems/next-permutation/
【题目】
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
【思路】
我们首先考虑特殊情况,即数组本来是逆序,那么将整个数组翻转即可。
一般情况呢,1234的下一个是1243,下一个是1324,下一个是1342,下一个是2134,下一个是2143……
仔细品品
发现规律了吗?
从后往前,找到第一个较小的数,即nums[i] < nums[i+1],将该数与其后元素中大于该数的最小元素进行替换,最后将后面的元素进行翻转。比如,1243,找到第一个较小的数2,替换后变成1342,翻转后得到结果1324。
【代码】
python版本
class Solution(object):
    def nextPermutation(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        ls = [nums[-1]]
        for i in range(len(nums) - 2, -1, -1):
            if nums[i] >= ls[-1]:
                ls.append(nums[i])
            else:
                # 大于nums[i]的最小元素
                for j in range(len(ls)):
                    if nums[i] < ls[j]:
                        tmp = ls.pop(j)
                        ls.insert(j, nums[i])
                        nums[i] = tmp
                        break
                # 其他元素,逆序
                for j in range(len(ls)):
                    nums[i+j+1] = ls[j]
                return 
        # 特殊情况,本身是最大的数    
        # 这种方式不行,必须是就地修改
        # nums = nums[::-1]
        for i in range(len(nums) / 2):
            nums[i], nums[-i-1] = nums[-i-1], nums[i]
C++版本
class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        vector<int> ls(1, nums.back());
        int i = 0;
        for (i = nums.size() - 2; i >= 0; i--) {
            if (nums[i] >= ls.back()) {
                ls.push_back(nums[i]);
            }else{
                // 大于nums[i]的最小元素,替换
                for (int j = 0; j < ls.size(); j++) {
                    if (nums[i] < ls[j]) {
                        int tmp = nums[i];
                        nums[i] = ls[j];
                        ls[j] = tmp;
                        break;
                    }
                }
                break;
            }
        }
        // 剩余元素,翻转
        for (int j = 0; j < ls.size(); j++) {
            nums[i + j + 1] = ls[j];
        }
    }
};
前一篇文章:【leetcode刷题】20T16-移除元素
给我好看