Leetcode 1013.将数组分成和相等的三个部分（Partition Array Into Three Parts With Equal Sum）

Leetcode 1013.将数组分成和相等的三个部分

1 题目描述（Leetcode题目链接）

  给你一个整数数组 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])$ 就可以将数组三等分。

输出：[0,2,1,-6,6,-7,9,1,2,0,1]
输出：true
解释：0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1

输入：[0,2,1,-6,6,7,9,-1,2,0,1]
输出：false

输入：[3,3,6,5,-2,2,5,1,-9,4]
输出：true
解释：3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4

提示：

• 3 <= A.length <= 50000
• -10^4 <= A[i] <= 10^4

2 题解

  首先判断数组总和是否能被3整除，如果不能那就说明无法分3段了。如果能，那除以3的结果就是分得的三个部分的结果，令$target=sum(A)/3$，从头遍历数组并记录前$i$项和，如果找到了第一部分和等于$target$，那就从下一个位置开始找，如果第二部分也找到了就直接返回True。不用管第三部分。

class Solution:
    def canThreePartsEqualSum(self, A: List[int]) -> bool:
        target = sum(A)
        length = len(A)
        if target%3 != 0:
            return False
        target = target//3
        part1 = 0
        for i in range(length-2):
            part1 += A[i]
            if part1 == target:
                part2 = 0
                for j in range(i+1, length-1):
                    part2 += A[j]
                    if part2 == target:
                        return True
        return False