915. 分割数组(中等题)
给定一个数组 nums ,将其划分为两个连续子数组 left 和 right, 使得:
left 中的每个元素都小于或等于 right 中的每个元素。
left 和 right 都是非空的。
left 的长度要尽可能小。
在完成这样的分组后返回 left 的 长度 。
用例可以保证存在这样的划分方法
方法一:两次遍历
思路和算法
题目要求将数组 nums 划分为非空的两个连续子数组 left 和right,并且需要满足 left 中的每个元素都小于等于 right 中的每个元素,同时 left 的长度要尽可能的小。
left 中的每个元素都小于或等于right 中的每个元素,等价于left 中的最大值小于等于right 中的最小值。我们可以维护一个数组 minRight,设 nums 的长度为 n
然后我们从小到大去遍历 i,过程中维护一个变量 maxLeft,它的值等于 max{0}^{i}{nums}[i]
找到第一个满足{maxLeft} \le \textit{maxRight}[i+1]maxLeft≤maxRight[i+1] 的 ii 即为答案,此时 \textit{left}left 的长度为 i + 1i+1,因此答案需返回 i + 1i+1。
注意,由于 \textit{left}left 和 \textit{right}right 都是非空子数组,ii 的遍历范围应当是 [0, n-2][0,n−2]。但因为题目保证有解,所以 ii 在遍历到 n - 1n−1 之前一定可以找到答案。
class Solution {
public:
int partitionDisjoint(vector<int>& nums) {
int n = nums.size();
vector<int> minRight(n);
minRight[n - 1] = nums[n - 1];
for (int i = n - 2; i >= 0; i--) {
minRight[i] = min(nums[i], minRight[i + 1]);
}
int maxLeft = 0;
for (int i = 0; i < n - 1; i++) {
maxLeft = max(maxLeft, nums[i]);
if (maxLeft <= minRight[i + 1]) {
return i + 1;
}
}
return n - 1;
}
};