LeetCode 每日一题:分割数组

156 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第 14 天,点击查看活动详情

分割数组

原题地址

给定一个数组 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 <=10510^5
  • 0 <= nums[i] <=10610^6
  • 可以保证至少有一种方法能够按题目所描述的那样对 nums 进行划分。

思路分析

最初拿到题目的想法就是遍历数组,然后分别按照每个位置分离出来的 left 和 right,来判断 left 中的值是否都小于 right 中的值,而判断的方式第一反应就是数组的 some 或者 every 方法,看到提示中数组的长度可能会很长,那么一定会导致超时或者内存不足,因此需要考虑别的实现方法。

  1. 题目要求 left 的长度尽可能的小,而 right 中的每个值都大于 left 中的值;
  2. 假设 leftMaxNum = nums[0] 是 left 中的最大值,从左到右来记录 left 中的最大值;
  3. 定义 maxNum 为 [0, i] 的最大值;
  4. 遍历数组,如果 nums[i] 小于 left 中的最大值,那么可以将该值存入左侧数组 left 中,并且需要重新计算 left 中的最大值,因为题目要求 right 中的所有值都必须大于 left 中的所有值,并且将 pos 值向右移动到当前位置 i;
  5. 待数组遍历完成后,就可以找到分界值 pos,最后返回即可。

AC 代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var partitionDisjoint = function(nums) {
    let len = nums.length
    // [0, i]的最大值
    let maxNum = nums[0]
    // left 最大值
    let leftMaxNum = nums[0]
    // 分界点
    let pos = 0
    for (let i = 1; i < len; i++) {
        maxNum = Math.max(maxNum, nums[i])
        if (nums[i] < leftMaxNum) {
            leftMaxNum = maxNum
            pos = i
        }
    }
    return pos + 1
};

结果:

  • 执行结果: 通过

END