[CTCI] 最小调整有序

最小调整有序

题目描述

有一个整数数组，请编写一个函数，找出索引m和n，只要将m和n之间的元素排好序，整个数组就是有序的。注意：n-m应该越小越好，也就是说，找出符合条件的最短序列。

给定一个int数组A和数组的大小n，请返回一个二元组，代表所求序列的起点和终点。(原序列位置从0开始标号,若原序列有序，返回[0,0])。保证A中元素均为正整数。

测试样例：

[1,4,6,5,9,10],6

返回：[2,3]

先分别找到左边第一对逆序对与右边第一对逆序对，然后找出两个逆序对之间的最大值与最小值，然后分别向两边扩展边界到最小值与最大值。

class Rearrange {
public:
    int findEndOfLeft(vector<int> &A) {
        for (int i = 0; i < A.size(); ++i) {
            if (A[i] < A[i-1]) return i - 1;
        }
        return A.size() - 1;
    }
    int findStartOfRight(vector<int> &A) {
        for (int i = A.size() - 2; i >= 0; --i) {
            if (A[i] > A[i+1]) return i + 1;
        }
        return 0;
    }
    int shrinkLeft(vector<int> &A, int min_idx, int start) {
        int comp = A[min_idx];
        for (int i = start - 1; i >= 0; --i) {
            if (A[i] <= comp) return i + 1;
        }
        return 0;
    }
    int shrinkRight(vector<int> &A, int max_idx, int start) {
        int comp = A[max_idx];
        for (int i = start; i < A.size(); ++i) {
            if (A[i] >= comp) return i - 1;
        }
        return A.size() - 1;
    }
    vector<int> findSegment(vector<int> A, int n) {
        // write code here
        int end_left = findEndOfLeft(A);
        int start_right = findStartOfRight(A);
        int min_idx = end_left + 1;
        if (min_idx >= A.size()) return {0, 0};
        int max_idx = start_right - 1;
        for (int i = end_left; i <= start_right; ++i) {
            if (A[i] < A[min_idx]) min_idx = i;
            if (A[i] > A[max_idx]) max_idx = i;
        }
        int left_idx = shrinkLeft(A, min_idx, end_left);
        int right_idx = shrinkRight(A, max_idx, start_right);
        return {left_idx, right_idx};
    }
};