leetcode - 每日一题 - 915. 分割数组

131 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 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 <= 105
  • 0 <= 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;
    }
};

image.png

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;
    }
};