分割数组

203 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招系列活动-刷题打卡任务,点击查看活动详情

一、问题描述

给定一个数组 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 的长度要尽可能小left 中的每个元素都小于或等于 right 中的每个元素。\

1、使用辅助数组

以示例1为例进行分析,我们可以这样做:

  • 生成一个辅助数组,记录每一个位置到0这个区间的最小值。 image.png 如上图,我们计算出了[0,i](i > 0 && i < nums.length)每一段区间的最大值。
  • 从后往前遍历,更新left的长度 从后往前遍历,不断更新[i,nums.leng - 1]的最小值,最小值大于[0,i]区间的最大值时,需要更新left的长度为i

AC代码

 * @param {number[]} nums
 * @return {number}
 */
var partitionDisjoint = function(nums) {
    let increment = [],m = 0;
    for(let i = 0; i < nums.length; i++){
        m = Math.max(m,nums[i]);
        increment.push(m);
    }
    m = Infinity;
    let res = 0;
    for(let i = nums.length - 1; i > 0; i--){
        m = Math.min(m,nums[i]);
        if(m >= increment[i - 1]) res = i;
    }
    return res;
};

2、不使用辅助数组

上面的方法需要借助一个辅助数组,这样会造成额外的空间消耗,我们可以减少这样的消耗,对程序进行优化。

  • 从左往右遍历 for (let i = 1; i < nums.length; i++)
  • 不断更新最大值 max = Math.max(max, nums[i]);
  • 维护左边区间的最大值 因为题目要求left 的长度要尽可能小,所以不到迫不得已我们不应该增大左边区间的队伍,那么什么情况向不得不扩大左边区间呢?这时候我们可以看看另一个关键条件:left 中的每个元素都小于或等于 right 中的每个元素,也就是说在遇到num[i]<leftMax的时候,我们必须要将num[i]纳入左边区间。
if (nums[i] < leftMax) {
  leftMax = max;
  pos = i;
}

AC代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var partitionDisjoint = function (nums) {
  let max = nums[0];
  let leftMax = nums[0];
  let pos = 0;
  for (let i = 1; i < nums.length; i++) {
    max = Math.max(max, nums[i]);
    if (nums[i] < leftMax) {
      leftMax = max;
      pos = i;
    }
  }
  return pos + 1;
};