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).

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

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,1

思路：

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

比如[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 Solution {
public:
    void nextPermutation(vector<int>& nums) {
        if (nums.size() == 0) return;
        int st = nums.size() - 1;
        while (st > 0 && nums[st - 1] >= nums[st]) st--;
        std::reverse(nums.begin() + st, nums.end());
        if (st == 0) return;
        auto it = std::upper_bound(nums.begin() + st, nums.end(), nums[st - 1]);
        std::swap(nums[st - 1], *it);
    }
};