给定数组array和K值,返回连续子数组和>=K的最小子数组的长度,如果不存在则返回-1.
ps: 面试的时候面试官不太懂python,交流了几句我被绕进去了,以为算法写错了,过后再看好像自己的算法没有问题,如有问题,希望大佬指出.
# 双指针实现
class Solution:
def func(self, array, K):
length = len(array)
# 开始索引
index = 0
# 累加值
number = 0
res = float('inf')
for i in range(length):
if number > 0:
number += array[i]
else:
number = array[i]
index = i
# 累加值大于等于K,index右移找到最小满足的位置
if number >= K:
res = min(res, i - index + 1)
while index < i:
number -= array[index]
index += 1
if number >= K:
res = min(res, i - index + 1)
else:
break
return res if res != float('inf') else -1