LintCode 190: Next Permutation

LintCode 190: Next Permutation

题目描述

给定一个若干整数的排列，给出按正数大小进行字典序从小到大排序后的下一个排列。

如果没有下一个排列，则输出字典序最小的序列。

样例

左边是原始排列，右边是对应的下一个排列。

1,2,3 → 1,3,2

3,2,1 → 1,2,3

1,1,5 → 1,5,1

Fri Feb 24 2017

思路

先看一个例子：

8, 5, 3, 7, 6, 5, 4, 1

下一个排列应该是：

8, 5, 4, 1, 3, 5, 6, 7

看起来应该是要从后往前找，找到第一个降序的数字，因为如果一直都是升序的话，说明这一部分以及是全排列的最大情况了。第一个比它后一个数小的数，就是需要替换为比他大一点的那个数。在上面的例子中就是3.

找到3以后，需要把它替换成一个比它大一点的数，于是再从后往前找，找到第一个比它大的数，在上面的例子中，就是4左边是原始排列，右边是对应的下一个排列。

将3和4交换，由于后半部分刚才已经是最大的全排列了，所以需要翻转一下，变成最小的全排列。

整个过程大概如下：

8, 5, 3, 7, 6, 5, 4, 1

8, 5, 3, 7, 6, 5, 4, 1

8, 5, 3, 7, 6, 5, 4, 1

8, 5, 4, 7, 6, 5, 3, 1

8, 5, 4, 7, 6, 5, 3, 1

8, 5, 4, 1, 3, 5, 6, 7

考虑一些特殊情况，若数组长度小于等于1，则下一个排列就是它本身，所以直接返回。

若数组本身就是最大的全排列，则在第一次从前往后寻找第一个比它后一个数小的数时，找到最前面也找不到，于是就不需要找第二个数了，直接将整个数组翻转就好了。

代码

// 下一个排列
void nextPermutation(vector<int> &nums)
{
    if (nums.size() <= 1) return;
    vector<int>::iterator l = nums.end() - 2;
    for (; l >= nums.begin(); --l)
        if (*l < *(l + 1)) break;
    if (l >= nums.begin())
    {
        vector<int>::iterator r = nums.end() - 1;
        for (; r != l; --r)
            if (*r > *l) break;
        swap(*l, *r);
    }
    reverse(l + 1, nums.end());
}