题:
给你一个整数数组 A,只有可以将其划分为三个和相等的非空部分时才返回 true,否则返回 false。
形式上,如果可以找出索引 i+1 < j 且满足 A[0] + A[1] + … + A[i] == A[i+1] + A[i+2] + … + A[j-1] == A[j] + A[j-1] + … + A[A.length - 1] 就可以将数组三等分。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum
题解详见代码
class Solution:
def canThreePartsEqualSum(self, arr: List[int]) -> bool:
#双指针
s =sum(arr)
if s % 3: return False#如果和不能整除于3,表示无法成功划分
left = 0
right = len(arr) - 1
left_sum = arr[left]
right_sum = arr[right]
while left < right - 1:#right-1是为了保证能将列表划分为3部分
if left_sum == right_sum == s / 3:
return True
if left_sum != s / 3:
left += 1
left_sum += arr[left]
if right_sum != s / 3:
right -= 1
right_sum += arr[right]
return False
#如若13行和14行交换,16行和17行交换,边界判断会出现错误