Next Permutation -- LeetCode

Implement next permutation, which rearranges numbers into the lexicographically next greater permutation of numbers.数组

If such arrangement is not possible, it must rearrange it as the lowest possible order (ie, sorted in ascending order).spa

The replacement must be in-place, do not allocate extra memory.code

Here are some examples. Inputs are in the left-hand column and its corresponding outputs are in the right-hand column.
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1blog

思路:it

一个数最大的时候,它的各个位从左向右应该是从大到小的。咱们找到nums数组末尾从大到小的一段,记录下该段的第一个元素的位置st。io

好比[0, 2, 4, 5, 2, 1], 末尾从大到小的一段是[4, 2, 1],st是3。咱们将该段的元素反转,而后将st-1位置的元素与该段中第一个大于它的元素交换位置(若st为0则不用交换)。这里st-1位置的元素是4,该段中第一个大于它的值为5。因此最后结果是[0, 2, 5, 1, 2, 4]。class

 1 class Solution {
 2 public:
 3     void nextPermutation(vector<int>& nums) {
 4         if (nums.size() == 0) return;
 5         int st = nums.size() - 1;
 6         while (st > 0 && nums[st - 1] >= nums[st]) st--;
 7         std::reverse(nums.begin() + st, nums.end());
 8         if (st == 0) return;
 9         auto it = std::upper_bound(nums.begin() + st, nums.end(), nums[st - 1]);
10         std::swap(nums[st - 1], *it);
11     }
12 };
相关文章
相关标签/搜索