【LetMeFly】3904.最小稳定下标 II:前后缀分解 —— 附Python3行版
力扣题目链接:https://leetcode.cn/problems/smallest-stable-index-ii/
给你一个长度为n的整数数组nums和一个整数k。
对于每个下标i,定义它的不稳定值为max(nums[0..i]) - min(nums[i..n - 1])。
换句话说:
max(nums[0..i])表示从下标 0 到下标i的元素中的最大值。min(nums[i..n - 1])表示从下标i到下标n - 1的元素中的最小值。
如果某个下标i的不稳定值小于等于k,则称该下标为稳定下标。
返回最小的稳定下标。如果不存在这样的下标,则返回-1。
示例 1:
输入:nums = [5,0,1,4], k = 3
输出:3
解释:
- 在下标 0 处:
[5]中的最大值是 5,[5, 0, 1, 4]中的最小值是 0,因此不稳定值为5 - 0 = 5。 - 在下标 1 处:
[5, 0]中的最大值是 5,[0, 1, 4]中的最小值是 0,因此不稳定值为5 - 0 = 5。 - 在下标 2 处:
[5, 0, 1]中的最大值是 5,[1, 4]中的最小值是 1,因此不稳定值为5 - 1 = 4。 - 在下标 3 处:
[5, 0, 1, 4]中的最大值是 5,[4]中的最小值是 4,因此不稳定值为5 - 4 = 1。 - 这是第一个不稳定值小于等于
k = 3的下标,因此答案是 3。
示例 2:
输入:nums = [3,2,1], k = 1
输出:-1
解释:
- 在下标 0 处,不稳定值为
3 - 1 = 2。 - 在下标 1 处,不稳定值为
3 - 1 = 2。 - 在下标 2 处,不稳定值为
3 - 1 = 2。 - 这些值都不小于等于
k = 1,因此答案是-1。
示例 3:
输入:nums = [0], k = 0
输出:0
解释:
在下标 0 处,不稳定值为0 - 0 = 0,它小于等于k = 0。因此答案是 0。
提示:
1 <= nums.length <= 1050 <= nums[i] <= 1090 <= k <= 109
解题方法:前后缀分解
同3903.最小稳定下标 I:O(n^2)或O(n)的方法二,倒序遍历一遍n u m s numsnums数组,得到“后续最小值数组”m i n i minimini,其中m i n i [ i ] mini[i]mini[i]表示从下标i ii到下标n − 1 n-1n−1的最小值。
再从前到后遍历n u m s numsnums数组,同时维护一个遍历过程中的最大值M MM,若M − m i n i [ i ] ≤ k M-mini[i]\leq kM−mini[i]≤k,则直接返回下标i ii。
若遍历完成未返回则返回− 1 -1−1。
- 时间复杂度O ( l e n ( n u m s ) ) O(len(nums))O(len(nums))
- 空间复杂度O ( l e n ( n u m s ) ) O(len(nums))O(len(nums))
AC代码
C++
/* * @LastEditTime: 2026-09-05 08:26:32 */classSolution{public:intfirstStableIndex(vector<int>&nums,intk){intn=nums.size();vector<int>mini(n);mini.back()=nums.back();for(inti=n-2;i>=0;i--){mini[i]=min(mini[i+1],nums[i]);}for(inti=0,M=0;i<n;i++){M=max(M,nums[i]);if(M-mini[i]<=k){returni;}}return-1;}};Python
''' LastEditTime: 2026-09-05 08:35:17 '''importitertoolsclassSolution:deffirstStableIndex(self,nums:list[int],k:int)->int:mini=list(itertools.accumulate(nums[::-1],min))[::-1]maxi=list(itertools.accumulate(nums,max))returnnext((ifori,(M,m)inenumerate(zip(maxi,mini))ifM-m<=k),-1)Python也可以一行完成,只是可读性会很差。
Java
/* * @LastEditTime: 2026-09-05 08:49:55 */classSolution{publicintfirstStableIndex(int[]nums,intk){intn=nums.length;int[]mini=newint[n];mini[n-1]=nums[n-1];for(inti=n-2;i>=0;i--){mini[i]=Math.min(nums[i],mini[i+1]);}for(inti=0,M=0;i<n;i++){M=Math.max(M,nums[i]);if(M-mini[i]<=k){returni;}}return-1;}}Go
/* * @LastEditTime: 2026-09-05 08:45:09 */packagemainfuncfirstStableIndex(nums[]int,kint)int{n:=len(nums)mini:=make([]int,n)mini[n-1]=nums[n-1]fori:=n-2;i>=0;i--{mini[i]=min(mini[i+1],nums[i])}M:=0fori,t:=rangenums{M=max(M,t)ifM-mini[i]<=k{returni}}return-1}Rust
/* * @LastEditTime: 2026-09-05 08:55:35 */implSolution{pubfnfirst_stable_index(nums:Vec<i32>,k:i32)->i32{letn=nums.len();letmutmini=vec![0;n];mini[n-1]=nums[n-1];foriin(0..n-1).rev(){mini[i]=nums[i].min(mini[i+1]);}letmutM=0;foriin0..n{M=M.max(nums[i]);ifM-mini[i]<=k{returniasi32;}}-1}}同步发文于CSDN和我的个人博客,原创不易,转载经作者同意后请附上原文链接哦~
千篇源码题解已开源