持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第9天,点击查看活动详情
915. 分割数组
难度:中等
题目:
给定一个数组 nums ,将其划分为两个连续子数组 left 和 right, 使得:
left中的每个元素都小于或等于right中的每个元素。left和right都是非空的。left的长度要尽可能小。
在完成这样的分组后返回 left 的 长度 。
用例可以保证存在这样的划分方法。
示例 1:
输入: nums = [5,0,3,8,6]
输出: 3
解释: left = [5,0,3],right = [8,6]
示例 2:
输入: nums = [1,1,1,0,6,12]
输出: 4
解释: left = [1,1,1,0],right = [6,12]
提示:
2 <= nums.length <= 1050 <= nums[i] <= 106- 可以保证至少有一种方法能够按题目所描述的那样对
nums进行划分。
个人思路
1.两次遍历
题目要求将数组划分为两个非空的两个连续数组,要求left数组中所有元素小于right数组中所有元素。
等价于,left数组中最大值,小于等于right数组中的最小值。
新建一个数组 right_max,数组中存入right数组中后几个数的最小值,从n到1。
然后对数组从前遍历,计算前n个数的最大值。与right_min[i]做比较,找到第一个满足条件的数。left_max[i]<=right_min[i]。
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;
}
};
2.一次遍历
再优化。
class Solution {
public:
int partitionDisjoint(vector<int>& nums) {
int n = nums.size();
int leftMax = nums[0], leftPos = 0, curMax = nums[0];
for (int i = 1; i < n - 1; i++) {
curMax = max(curMax, nums[i]);
if (nums[i] < leftMax) {
leftMax = curMax;
leftPos = i;
}
}
return leftPos + 1;
}
};