Keshawn_lu's Blog

Leetcode 915. 分割数组

字数统计: 499阅读时长: 2 min
2022/10/24 Share

题目简介:

给定一个数组 nums ,将其划分为两个连续子数组 leftright, 使得:

  • left 中的每个元素都小于或等于 right 中的每个元素。
  • leftright 都是非空的。
  • left 的长度要尽可能小。

在完成这样的分组后返回 left长度

用例可以保证存在这样的划分方法。

示例 1:

1
2
3
输入:nums = [5,0,3,8,6]
输出:3
解释:left = [5,0,3],right = [8,6]

示例 2:

1
2
3
输入:nums = [1,1,1,0,6,12]
输出:4
解释:left = [1,1,1,0],right = [6,12]

提示:

  • 2 <= nums.length <= 10^5
  • 0 <= nums[i] <= 10^6
  • 可以保证至少有一种方法能够按题目所描述的那样对 nums 进行划分。

思路:

方法一:

每次从后往前遍历找到第一个不满足要求的元素下标,即(left_max_num > nums[right])

因此我们必须将把nums[right]加入到left数组中,同时更新left数组的最大值。

tip:

  • 使用max_element()来寻找范围内的最大值

方法二:

从前往后一次遍历,当nums[i] < left_max时,说明需要将nums[i]加入至left数组中,同时更新当前left数组的最大值,保存nums[i]的位置下标,最后返回res_pos + 1即可。

代码如下:

方法一:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
class Solution {
public:

int solve(int num, vector<int>& nums, int pos){

for(int i = nums.size() - 1; i >= pos + 1; i--){

if(num > nums[i])
return i;
}

return -1;
}

int partitionDisjoint(vector<int>& nums) {

int max_num = nums[0];
for(int i = 0; i < nums.size(); i++){

int pos = solve(max_num, nums, i);
if(pos == -1)
return i + 1;
else{

i = pos - 1;
max_num = *max_element(nums.begin(), nums.begin() + i + 1);
}

}

return nums.size();
}
};

方法二:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution {
public:
int partitionDisjoint(vector<int>& nums) {

int now_max = nums[0];
int left_max = nums[0];
int res_pos = 0;


for(int i = 0; i < nums.size(); i++){

now_max = max(now_max, nums[i]);

if(nums[i] < left_max){ //说明需要将nums[i]加入left数组中

res_pos = i;
left_max = now_max;
}
}

return res_pos + 1;
}
};
CATALOG
  1. 1. 题目简介:
  2. 2. 思路:
    1. 2.1. 方法一:
    2. 2.2. 方法二:
  3. 3. 代码如下:
    1. 3.1. 方法一:
    2. 3.2. 方法二: