Leetcode 1020：将数组分成和相等的三个部分（超详细的解法！！！）

版权声明：本文为博主原创文章，未经博主允许不得转载。有事联系：[email protected] https://blog.csdn.net/qq_17550379/article/details/88825818

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

示例 1：

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

示例 2：

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

示例 3：

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

提示：

1. 3 <= A.length <= 50000
2. -10000 <= A[i] <= 10000

解题思路

这个问题非常简单，我们首先计算sum(A)%3==0，如果不是的话，自然就不能拆分为3份。然后接着判断是不是有3份small//3即可。

class Solution:
    def canThreePartsEqualSum(self, A: List[int]) -> bool:
        n = len(A)
        total = sum(A)
        if total%3 != 0:
            return False
        
        p = total//3
        tmp, t = 0, 0
        for i in range(n):
            tmp += A[i]
            if tmp == p:
                t = i
                break
        tmp = 0
        for j in range(t+1, n):
            tmp += A[j]
            if tmp == p:
                return sum(A[j+1:]) == p
        return False

更加简洁的写法

class Solution:
    def canThreePartsEqualSum(self, A: List[int]) -> bool:
        total, n = sum(A), len(A)
        if total%3 != 0:
            return False
        
        p = total//3
        tmp, t, res = 0, 0, 0
        for i in range(n):
            tmp += A[i]
            if tmp == p:
                tmp = 0
                res += 1
        return res >= 3

注意我们最后需要判断res>=3，为什么会出现>3呢？对于这个例子[1, -1, 1, -1, 1, -1, 1, -1]。

reference:

扫描二维码关注公众号，回复： 5733651 查看本文章

https://leetcode.com/problems/partition-array-into-three-parts-with-equal-sum/discuss/260885/C%2B%2B-6-lines-O(n)-target-sum

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题，希望大家指出！！！