Leetcode 1053. 交换一次的先前排列

题目

给你一个正整数的数组 A（其中的元素不一定完全不同），请你返回可在 一次交换（交换两数字 A[i] 和 A[j] 的位置）后得到的、按字典序排列小于 A 的最大可能排列。

如果无法这么操作，就请返回原数组。

示例 1：

输入：arr = [3,2,1]
输出：[3,1,2]
解释：交换 2 和 1

示例 2：

输入：arr = [1,1,5]
输出：[1,1,5]
解释：已经是最小排列

示例 3：

输入：arr = [1,9,4,6,7]
输出：[1,7,4,6,9]
解释：交换 9 和 7

示例 4：

输入：arr = [3,1,1,3]
输出：[1,3,1,3]
解释：交换 1 和 3

思路

• 从后往前找， 找到第一个相邻逆序对， 记录下标为pos_i
• 从这个位置向后找比arr[pos_i]小的且最为接近arr[pos_i]的
• 交换之

代码

class Solution {
public:
    vector<int> prevPermOpt1(vector<int>& arr) {
        int size = arr.size();

        if (size == 1) return arr;
        else {
            int pos_i;

            for (pos_i = size - 2; pos_i >= 0; --pos_i) {
                if (arr[pos_i] > arr[pos_i + 1]) break;
            }

            if (pos_i == -1) return arr;

            int pos_j = pos_i + 1;

            for (int j = pos_i + 1; j < size; ++j) {
                if (arr[j] < arr[pos_i] && arr[j] > arr[pos_j]) pos_j = j;
            }

            swap(arr[pos_i], arr[pos_j]);
            return arr;
        }
    }
};