【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;
    }
}