第二道很明显用双指针更简单一点吧,i指向第一个数,j指向最后一个数,两边往中间移就好了,写了一下,这么写的好处是不用讨论你博客中写的那些特殊情况,都包含在这段代码里了,按我的逻辑是没有问题的,有问题的话请指教!
def equal_3(arr):
# 判断特殊输入
if not arr or len(arr) < 3:
return False
left = 0 # 表示分成三组之后,第一组的最后元素的下标
right = len(arr) - 1 # 表示分成三组之后,第三组的第一个元素下标
sum_l = arr[0] # 第一组的和
sum_r = arr[len(arr) - 1] #第三组的和
# 当最后left=right时,必然不存在满足的分法
while left < right:
# 如果第一组的和小于第三组,left右移,sum_l增加;如果第一组的和大于第三组,right左移,sum_r增加
# 相等的情况时就判断中间的那组和是否和这两组和相同,相同则返回True,不同的话因为可能会存在[1,2,3,6,3,1,2]
# 这种情况,所有左右指针同时移
if sum_l < sum_r:
left += 1
sum_l += arr[left]
elif sum_l > sum_r:
right -= 1
sum_r += arr[right]
else:
if sum(arr[left+1:right]) == sum_l:
return True
else:
left += 1
sum_l += arr[left]
right -= 1
sum_r += arr[right]
return False