来源:力扣(LeetCode)
描述:
给定一个数组 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 <= 105
- 0 <= nums[i] <= 106
- 可以保证至少有一种方法能够按题目所描述的那样对 nums 进行划分。
方法一:一次遍历
思路和算法
假设我们预先规定了一个 left 的划分,其最大值为 maxLeft,划分位置为 leftPos,表示 nums[0,leftPos] 都属于 left。如果 leftPos 右侧所有元素都大于等于它,那么该划分方案是合法的。
但如果我们找到 nums[i],其中 i > leftPos,并且 nums[i] < maxLeft,那么意味着 leftPos 作为划分位置是非法的,需要更新 leftPos = i,以及
因此,我们首先初始化 maxLeft = nums[0], leftPos = 0,然后在 [1, n−2] 范围内从小到大遍历 i,过程中维护一个变量 curMax,它的值是
此时如果有 nums[i] < maxLeft,就按照上述方法更新。最终遍历结束时,答案就是 leftPos + 1。
代码:
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;
}
};
执行用时:124 ms, 在所有 C++ 提交中击败了68.94%的用户
内存消耗:92 MB,在所有 C++ 提交中击败了61.92%的用户
复杂度分析
时间复杂度:O(n),其中 n 是 nums 的长度。从前到后遍历 i 的时间复杂度为 O(n)。
空间复杂度:O(1)。算法只使用了常数个变量。
方法二:两次遍历
思路和算法
题目要求将数组 nums 划分为非空的两个连续子数组 left 和 right,并且需要满足 left 中的每个元素都小于等于 right 中的每个元素,同时 left 的长度要尽可能的小。
left 中的每个元素都小于或等于 right 中的每个元素,等价于 left 中的最大值小于等于 right 中的最小值。我们可以维护一个数组 minRight,设 nums 的长度为 n,那么
然后我们从小到大去遍历 i,过程中维护一个变量 maxLeft,它的值等于 。找到第一个满足 maxLeft ≤ maxRight[i+1] 的 i 即为答案,此时 left 的长度为 i + 1,因此答案需返回 i + 1。
注意,由于left 和 right 都是非空子数组,i 的遍历范围应当是 [0, n - 2]。但因为题目保证有解,所以 i 在遍历到 n - 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;
}
};
执行用时:132 ms, 在所有 C++ 提交中击败了51.10%的用户
内存消耗:96 MB, 在所有 C++ 提交中击败了37.48%的用户
复杂度分析
时间复杂度:O(n),其中 n 是 nums 的长度。求解 minRight 的时间复杂度是 O(n),从前到后遍历 i 的时间复杂度也为 O(n)。
空间复杂度:O(n),其中 n 是 nums 的长度。使用辅助数组 minRight 的空间开销为 O(n)。
author:LeetCode-Solution