贪心题目：交换一次的先前排列

题目

标题和出处

标题：交换一次的先前排列

出处：1053. 交换一次的先前排列

难度

5 级

题目描述

要求

给定一个正整数数组 $\text{arr}$（其中的元素可能重复），返回可在一次交换（交换两数字 $\text{arr[i]}$ 和 $\text{arr[j]}$ 的位置）后得到的按字典序排列小于 $\text{arr}$ 的最大可能排列。如果无法得到字典序更小的排列，返回原数组。

示例

示例 1：

输入：$\text{arr = [3,2,1]}$
输出：$\text{[3,1,2]}$
解释：交换 $\text{2}$ 和 $\text{1}$。

示例 2：

输入：$\text{arr = [1,1,5]}$
输出：$\text{[1,1,5]}$
解释：已经是最小排列。

示例 3：

输入：$\text{arr = [1,9,4,6,7]}$
输出：$\text{[1,7,4,6,9]}$
解释：交换 $\text{9}$ 和 $\text{7}$。

数据范围

• $\text{1}\le \text{arr.length}\le {\text{10}}^{\text{4}}$
• $\text{1}\le \text{arr[i]}\le {\text{10}}^{\text{4}}$

解法

思路和算法

用 $n$ 表示数组 $\text{arr}$ 的长度。对于 $0\le i<j<n$，考虑交换 $\text{arr}[i]$ 和 $\text{arr}[j]$ 之后的字典序变化。

• 如果 $\text{arr}[i]\le \text{arr}[j]$，则交换 $\text{arr}[i]$ 和 $\text{arr}[j]$ 之后，下标 $i$ 处的元素值不变或增加，因此字典序不可能更小。

• 如果 $\text{arr}[i]>\text{arr}[j]$，则交换 $\text{arr}[i]$ 和 $\text{arr}[j]$ 之后，下标 $i$ 处的元素值减少，因此字典序更小。

因此需要找到两个下标 $\text{left}$ 和 $\text{right}$，满足 $\text{left}<\text{right}$ 且 $\text{arr}[\text{left}]>\text{arr}[\text{right}]$，才能在交换 $\text{arr}[\text{left}]$ 和 $\text{arr}[\text{right}]$ 之后得到字典序更小的排列。

为了得到字典序小于原始数组的最大可能排列，应使用贪心策略找到上述下标 $\text{left}$ 和 $\text{right}$。

1. 寻找 $\text{left}$ 时，应寻找符合上述要求的最大下标。理由如下：当 $\text{left}<\text{right}$ 且 $\text{arr}[\text{left}]>\text{arr}[\text{right}]$ 时，假设存在 ${\text{left}}^{\prime }$ 满足 ${\text{left}}^{\prime }<\text{left}$ 且 $\text{arr}[{\text{left}}^{\prime }]>\text{arr}[\text{right}]$，则交换 $\text{arr}[\text{left}]$ 和 $\text{arr}[\text{right}]$ 与交换 $\text{arr}[{\text{left}}^{\prime }]$ 和 $\text{arr}[\text{right}]$ 都可以得到字典序更小的排列，其中交换 $\text{arr}[{\text{left}}^{\prime }]$ 和 $\text{arr}[\text{right}]$ 得到的排列的字典序更小，一定不是可能的最大排列。寻找 $\text{left}$ 的可行做法是寻找满足 $\text{arr}[i]>\text{arr}[i+1]$ 的最大下标作为 $\text{left}$，此时的 $\text{left}$ 为符合上述要求的最大下标。

2. 在确定 $\text{left}$ 之后，记 $\text{leftNum}=\text{arr}[\text{left}]$，寻找 $\text{right}$ 时，应在数组 $\text{arr}$ 的下标大于 $\text{left}$ 的所有元素中寻找小于 $\text{arr}[\text{left}]$ 的最大元素所在的最小下标作为 $\text{right}$。理由如下：记 $\text{rightNum}=\text{arr}[\text{right}]$，为了通过交换 $\text{arr}[\text{left}]$ 和 $\text{arr}[\text{right}]$ 得到字典序更小的排列，必须满足 $\text{arr}[\text{left}]>\text{arr}[\text{right}]$，在下标大于 $\text{left}$ 的所有元素中，下标小于 $\text{right}$ 的所有元素都小于 $\text{rightNum}$，下标大于 $\text{right}$ 的所有元素都小于等于 $\text{rightNum}$，如果与 $\text{arr}[\text{left}]$ 交换的元素在下标 $\text{right}$ 的左侧则交换后位于下标 $\text{left}$ 处的元素一定小于 $\text{rightNum}$，如果与 $\text{arr}[\text{left}]$ 交换的元素在下标 $\text{right}$ 的右侧则交换后位于下标 $\text{right}$ 处的元素一定小于 $\text{leftNum}$，因此如果与 $\text{arr}[\text{left}]$ 交换的元素在下标 $\text{right}$ 的左侧或右侧则一定不是可能的最大排列。

具体做法如下。

1. 反向遍历数组 $\text{arr}$ 并找到满足 $\text{arr}[i]>\text{arr}[i+1]$ 的最大下标 $i$ 作为下标 $\text{left}$。如果不存在这样的下标 $i$，则记 $\text{left}=-1$。

2. 当 $\text{left}\ge 0$ 时，从下标 $\text{left}+1$ 开始正向遍历数组 $\text{arr}$ 并找到小于 $\text{arr}[\text{left}]$ 的最大元素的最小下标作为 $\text{right}$，然后交换 $\text{arr}[\text{left}]$ 和 $\text{arr}[\text{right}]$。

3. 经过上述操作之后得到的 $\text{arr}$ 即为答案。

代码

class Solution {
    public int[] prevPermOpt1(int[] arr) {
        int length = arr.length;
        int left = -1;
        for (int i = length - 2; i >= 0 && left < 0; i--) {
            if (arr[i] > arr[i + 1]) {
                left = i;
            }
        }
        if (left >= 0) {
            int right = left + 1;
            int rightMax = arr[left + 1];
            for (int i = left + 2; i < length; i++) {
                if (arr[i] < arr[left] && arr[i] > rightMax) {
                    right = i;
                    rightMax = arr[i];
                }
            }
            int temp = arr[left];
            arr[left] = arr[right];
            arr[right] = temp;
        }
        return arr;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $\text{arr}$ 的长度。最多需要遍历数组两次。

• 空间复杂度：$O(1)$。