[LeetCode] 031. Next Permutation (Medium) (C++/Python)

索引:[LeetCode] Leetcode 题解索引 (C++/Java/Python/Sql)
Github: https://github.com/illuz/leetcode


031. Next Permutation (Medium)

链接

题目:https://oj.leetcode.com/problems/next-permutation/
代码(github):https://github.com/illuz/leetcode

题意

求一个序列的下一个排列。

分析

可以用 STL 里的 ‘next_permutation’ 偷懒。

具体算法是:

首先,从最尾端开始往前寻找两个相邻的元素,令第一个元素是 i,第二个元素是 ii,且满足 i<ii
然后,再从最尾端开始往前搜索,找出第一个大于 i 的元素,设其为 j;
然后,将 i 和 j 对调,再将 ii 及其后面的所有元素反转。

代码

C++:

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

		int idx = num.size() - 2;

		// 1. find out the last wrong order
		while (idx >= 0 && num[idx] >= num[idx + 1])
			idx--;

		// 2. swap
		if (idx >= 0) {
			int i = idx + 1;
			while (i < num.size() && num[i] > num[idx])
				i++;
			swap(num[i - 1], num[idx]);
		}

		// 3. reverse
		reverse(num.begin() + idx + 1, num.end());
	}
};


Python:

class Solution:
    # @param num, a list of integer
    # @return nothing (void), do not return anything, modify num in-place instead.
    def nextPermutation(self, num):
        if not len(num):
            return

        idx = len(num) - 2

        # 1. find out the last wrong order
        while idx >= 0 and num[idx] >= num[idx + 1]:
            idx -= 1

        # 2. swap
        if idx >= 0:
            i = idx + 1
            while i < len(num) and num[i] > num[idx]:
                i += 1
            num[i - 1], num[idx] = num[idx], num[i - 1]

        # 3. reverse
        left, right = idx + 1, len(num) - 1
        while left <= right:
            num[left], num[right] = num[right], num[left]
            left += 1
            right -= 1


  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值