Leetcode 每日一题——31. 下一个排列

31. 下一个排列

实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

必须原地修改,只允许使用额外常数空间。

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
在这里插入图片描述
这道题在做之前首先要搞懂什么是字典序(不知道的请出门左拐百度),为了找到字典序的下一个排列,要分成两步处理,首先从尾节点开始找第一个降序节点的前一个节点(就是降序那个点),然后再从尾节点开始遍历,找到第一个比降序那个点大的并进行交换,然后将该节点后续节点按升序排列,具体的代码如下(比较啰嗦的C++版,好久之前写的,我自己现在都看不懂我当时是怎么想的):

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        if(!nums.size()) return;
        int numSize=nums.size();

        int index=findPosition(nums);
        
        if(index==-1)
        {
            reveseNums(nums,0,numSize-1);
            return;
        }
        int i;
        for(i=index+1;i<nums.size();i++)
        {
            if(nums[index]>=nums[i])
            {
                
                swap(nums[index],nums[i-1]);
                
                reveseNums(nums,index+1,numSize-1);
                
                break;
            }
        }
        if(i==numSize)
        {
            
            swap(nums[index],nums[numSize-1]);
            reveseNums(nums,index+1,numSize-1);
        }
        return;
    }
    int findPosition(vector<int>& nums)
    {
        if(!nums.size()) return -1;
        int n=nums.size();
        int tmp=nums[n-1];
        int index=n-1;
        int i;
        for(i=n-1;i>=0;i--)
        {
            
            if(nums[i]<tmp)
            {
                index=i;
                break;
            }
            else
            {
       
                tmp=nums[i];
            }
        }
        if(i==-1) return -1;
        
        return index;
    }
    int reveseNums(vector<int>& nums,int left, int right)
    {
        if(!nums.size()) return -1;
        
        while(left<right)
        {
            int tmp=nums[left];
            nums[left]=nums[right];
            nums[right]=tmp;
            left++;
            right--;
        }
        return 0;
    } 
};

运行效果(渣渣ε=(´ο`*)))唉):
Leetcode 每日一题——31. 下一个排列_第1张图片
相同思路Python代码如下:

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        if len(nums)<=1:
            return
        def findPosition(nums):
            right=len(nums)-1
            while(right-1>=0):
                if nums[right] > nums[right-1]:
                    break
                right-=1
            return right

        def reversList(nums,left,right):
            while(left<right):
                nums[left],nums[right]=nums[right],nums[left]
                left+=1
                right-=1
        
        index=findPosition(nums)
        
        if index==0:
            reversList(nums,0,len(nums)-1)
        else:
            tmpNum=nums[index-1]
            index_2=len(nums)-1
            while(index_2-1>=0):
                    if nums[index_2] > tmpNum:
                        break
                    index_2-=1
            nums[index-1],nums[index_2]=nums[index_2],nums[index-1]
            reversList(nums,index,len(nums)-1)

运行效果(还是比较一般):
Leetcode 每日一题——31. 下一个排列_第2张图片

来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/next-permutation

你可能感兴趣的:(数据结构和算法,leetcode,数据结构,算法,python)