LintCode : 下一个排列

LintCode : 下一个排列

题目

给定一个整数数组来表示排列，找出其之后的一个排列。

样例

给出排列[1,3,2,3]，其下一个排列是[1,3,3,2]

给出排列[4,3,2,1]，其下一个排列是[1,2,3,4]

思路

1、从右向左找到第一个递增数列；
2、找到递增数列中最小的比递增数列前一个数大的数；
3、交换这两个数的位置，将递增数列逆序；

例：146521
1、6521逆序递增，5为递增数列中大于4的数中最小的
2、交换位置变为156421
3、递增数列逆序151246
得146521的下一个排列为151246.

（参考产生下一个排列数算法）

代码

vector<int> nextPermutation(vector<int> &nums) {
    if(nums.size() == 1 || nums.size() == 0)
        return nums;
    if(nums.size() == 2) {
        swap(nums[0], nums[1]);
        return nums;
    }
    for(int i = nums.size() - 1; i > 0; i--) {
        if(nums[i] > nums[i - 1]) {
            for(int j = nums.size() - 1; j >= i; j--)
                if(nums[j] > nums[i - 1]) {
                    swap(nums[i - 1], nums[j]);
                    break;
                }
            for(int j = 0; j < (nums.size() - i) / 2; j++)
                swap(nums[i + j], nums[nums.size() - j - 1]);
            return nums;
        }
    }
    for(int j = 0; j < nums.size() / 2; j++)
        swap(nums[j], nums[nums.size() - j - 1]);
    return nums;
}