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
}