关注
第二道很明显用双指针更简单一点吧,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
查看原帖
点赞 6
相关推荐

点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 你认为小厂实习有用吗? #
13733次浏览 186人参与
# 面试官是我前女友 #
106890次浏览 731人参与
# 实习生的蛐蛐区 #
39797次浏览 329人参与
# 当你面对裁员会如何? #
275921次浏览 2438人参与
# 计算机有哪些岗位值得去? #
12816次浏览 133人参与
# lastday知无不言 #
56850次浏览 460人参与
# 在职场上,你最讨厌什么样的同事 #
14421次浏览 150人参与
# 推荐一首陪你工作的歌吧 #
14002次浏览 97人参与
# 说说你知道的学历厂 #
28110次浏览 178人参与
# 你找工作的时候用AI吗? #
14783次浏览 192人参与
# 下班后的时间你怎么安排 #
7395次浏览 111人参与
# 哪一瞬间觉得自己长大了 #
7212次浏览 166人参与
# 携程求职进展汇总 #
559402次浏览 4258人参与
# 面试尴尬现场 #
23619次浏览 163人参与
# 工作后会跟朋友渐行渐远吗 #
30031次浏览 216人参与
# 中核求职进展汇总 #
20127次浏览 152人参与
# 社会教会你的第一课 #
29184次浏览 394人参与
# 多益网络工作体验 #
49626次浏览 280人参与
# 虾皮求职进展汇总 #
244159次浏览 1797人参与
# 神州信息工作体验 #
15996次浏览 75人参与