力扣915(java&python)-分割数组(中等)
题目:
给定一个数组 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 进行划分。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/partition-array-into-disjoint-intervals
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
一、前缀最大值+后缀最小值:只需要保证前缀的最大值小于等于后缀的最小值。
先从后往前遍历,找到数组的所有后缀的最小值在min数组中,min[i] = x :下标范围在[i, n-1]中nums[i]的最小值x。然后再从前往后遍历数组,统计出数组前缀的最大值 max
,当遍历到某个位置时,如果数组前缀最大值小于等于数组后缀最小值,那么当前位置就是划分的分界点,然后返回长度(当前下标i + 1)。
java代码:
1 class Solution { 2 public int partitionDisjoint(int[] nums) { 3 int n = nums.length; 4 int[] min = new int[n]; 5 min[n-1] = nums[n-1]; 6 //从后往前遍历,取得后缀的最大值 7 for(int i = n-2; i >= 0; i--){ 8 min[i] = Math.min(nums[i], min[i+1]); 9 } 10 for(int i = 0, max = 0; i < n-1; i++){ 11 //从前往后遍历,取得前缀的最大值 12 max = Math.max(max, nums[i]); 13 if(max <= min[i+1]){ 14 return i + 1; 15 } 16 } 17 return -1; 18 } 19 }
python3代码:
1 class Solution: 2 def partitionDisjoint(self, nums: List[int]) -> int: 3 n = len(nums) 4 minright = [0] * n 5 minright[n-1] = nums[n-1] 6 for i in range(n-2, 0 , -1): 7 minright[i] = min(nums[i], minright[i+1]) 8 maxleft = 0 9 for i in range(0, n): 10 maxleft = max(maxleft, nums[i]) 11 if maxleft <= minright[i + 1]: 12 return i + 1 13 return -1
二、一次遍历
参考@爪哇缪斯:https://leetcode.cn/problems/partition-array-into-disjoint-intervals/solution/zhua-wa-mou-si-tu-jie-leetcode-by-muse-7-omut/
left 中的每个元素都小于或等于 right 中的每个元素,对于数组left,需要一个变量leftMax来保存数组left中的最大数字,且设leftMax初始值就为nums[0],需要一个变量max来保存当前的最大值,max的初始值也设为nums[0],然后从 i =1开始往后遍历nums数组中的每个数字,当发现遍历的这个数字nums[i]大于leftMax的时候,则表示这个数字暂时可能不属于数组left,更新一下最大值max。当发现遍历的这个数字nums[i]小于leftMax的时候,则可以判断出nums[i]一定是属于数组left的,则移动index到当前的位置 i,更新leftMax的值为max,当我们遍历完所有的nums数组中的数字之后,index指向的位置就是数组left的最后一个元素的位置。那么数组left的长度就等于index + 1。
请见下图所示:
java代码:
1 class Solution { 2 public int partitionDisjoint(int[] nums) { 3 int n = nums.length; 4 int leftMax = nums[0], max = nums[0], index = 0; 5 for(int i = 1; i < n; i++){ 6 if(leftMax > nums[i]){ 7 index = i; 8 leftMax = max; 9 }else{ 10 max = Math.max(max, nums[i]); 11 } 12 } 13 return index + 1; 14 } 15 }
小知识:
python中range():
range(n-1, 0, -1):从列表的下标为 i 的元素开始,倒序取到下标为0的元素(但是不包括下标为0元素)
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)