LeetCode 862. 和至少为 K 的最短子数组
思路: 采用滑动窗口
注意点
- 左边移动的时候,如果去掉的数是负数,那么右边游标要做回滚;
- 右边的游标移动时,出现sum<=0,说明此次的计算没有意义,不要再进行下去;
代码
func shortestSubarray(A []int, K int) int {
lA := len(A)
sum := 0
j := 0
minL := -1
for i := 0; i < lA; i++ {
if i >= 1 {
sum -= A[i-1]
// 回滚, 如果删的是负数
if A[i-1] < 0 {
for ; ; {
if j >i {
sum -= A[j-1]
j --
if sum >= K {
min := j - i
if minL == -1 || (minL != -1&& min < minL) {
minL = min
}
}
} else {
break
}
}
}
}
// 此时够的,那么计算最小
if sum >= K {
min := j - i
if minL == -1 || (minL != -1&& min < minL) {
minL = min
}
continue
}
for {
if j >= lA || sum < 0 {
break
}
sum += A[j]
j ++
if sum >= K {
min := j - i
if minL == -1 || (minL != -1&& min < minL) {
minL = min
}
break
}
}
}
return minL
}