LeetCode 31 Next Permutaion

173 阅读1分钟

LeetCode 31 Next Permutaion

思路

  1. 从右往左遍历,找到第一个nums[i] < nums[i+1]的位置i
  2. 从右往左遍历,找到第一个大于nums[i]的元素的位置j
  3. 交换nums[i]与nums[j]
  4. 翻转从位置i+1开始(包括),之后的元素。

代码

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