更多精彩內(nèi)容，請關(guān)注【力扣簡單題】。

題目

難度：★★☆☆☆
類型：數(shù)組

給定一個整數(shù)數(shù)組 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]) 就可以將數(shù)組三等分。

提示
3 <= A.length <= 50000
-10000 <= A[i] <= 10000

示例

示例 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. 判斷數(shù)組和s是否是3的倍數(shù)，如果不是，直接返回False；

2. 獲得每一部分的和avg=s/3，定義左右指針left=0和right=len(A)-1，左右部分和left_s=right_s=0；

3. 首先從左到右開始遍歷，查看遍歷到什么時候左段數(shù)組和為avg，如果沒有發(fā)現(xiàn)，則返回False；

4. 從右到左遍歷數(shù)組，查看何時右段數(shù)組和為avg，如果沒有發(fā)現(xiàn)，返回False；

5. 上述步驟都執(zhí)行通過，說明左段和右段和均為avg，中段也是avg，返回True。

class Solution:
    def canThreePartsEqualSum(self, A):
        """
        :param A: List[int]
        :return: bool
        """
        s = sum(A)

        if s % 3 != 0:
            return False

        avg = int(s / 3)

        left, right = 0, len(A)-1
        left_s, right_s = 0, 0

        while left_s != avg:
            if left > right:
                return False
            left_s += A[left]
            left += 1

        while right_s != avg:
            if left > right:
                return False
            right_s += A[right]
            right -= 1

        return True

如有疑問或建議，歡迎評論區(qū)留言~