【力扣LeetCode】31 下一个排列

版权声明:本文为原创博客,未经允许,请勿转载。 https://blog.csdn.net/u013095333/article/details/90815933

题目描述(难度中)

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

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

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

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

链接

https://leetcode-cn.com/problems/next-permutation/

思路

从后面往前面找,找到第一个前面的数小于后面的数的情况,确定该位置的数为需要调换到后面的数,将确定要调换的数与从后往前第一个大于该数的数互换,互换之后对确定的位置后面的数进行从小到大排序即可。
例如:3124321
确定需要调换的位置为3124321
从后往前找到3124321与2进行位置互换得到3134221
然后对3134221之后的数字4221进行从小到大排序,这里可以排序,也可以直接前后互换,前后互换复杂度显然更低
详细描述可参考:
https://leetcode-cn.com/problems/next-permutation/solution/xia-yi-ge-pai-lie-by-leetcode/

代码

最简洁的代码:

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

思路不成熟时,分情况讨论的算法:

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        bool flag = true;
        for(int i = nums.size()-1; i >= 0; i--){
        	for(int j = nums.size()-1; j > i; j--){
        		if(nums[i] < nums[j]){
        			swap(nums[i], nums[j]);
        			flag = false;
        			sort(nums.begin()+i+1, nums.end());
        			break;
        		}
        	}
			if(!flag){
				break;
			}
        }
        if(flag){
        	for(int i = 0; i < nums.size()/2; i++){
        		swap(nums[i], nums[nums.size()-1-i]);
        	}
        }
    }
};

猜你喜欢

转载自blog.csdn.net/u013095333/article/details/90815933