【LeetCode】31. 下一个排列

题目描述

链接：下一个排列
实现获取下一个排列的函数，算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。

必须原地修改，只允许使用额外常数空间。

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

解答思路

主要流程

1. 找出最大的数组索引k满足 num[k] < num[k + 1]，如果 k 没找到直接反转整个字符串；
2. 再找出最大索引j满足 num[j] > num[k]；
3. 交换 num[j] 和 num[k]；
4. 反转k + 1以后的字符串 。

因为都是找最大的索引k和j，所以我们从数组结尾开始找就可以，当找到的时候直接break就说明找到索引。

代码

public class Solution {
    public void nextPermutation(int[] num) {
        if (num == null || num.length == 0) {
            return;
        }
        int index1 = -1;
        for (int i = num.length - 2; i >= 0; i--) {
            if (num[i] < num[i + 1]) {
                index1 = i;
                break;
            }
        }
        if (index1 == -1) {
            reverse(num, 0, num.length - 1);
            return;
        }
        
        int index2 = -1;
        for (int i = num.length - 1; i > index1; i--) {
            if (num[i] > num[index1]) {
                index2 = i;
                break;
            }
        }
        swap(num, index1, index2);
        reverse(num, index1 + 1, num.length - 1);
    }
    
    public void reverse(int[] num, int i, int j) {
        while (i < j) {
            swap(num, i++, j--);
        }
    }
    
    public void swap(int[] num, int i, int j) {
        int temp = num[i];
        num[i] = num[j];
        num[j] = temp;
    }
}