LeetCode刷题之1013. 将数组分成和相等的三个部分

LeetCode刷题之1013. 将数组分成和相等的三个部分

我不知道将去向何方，但我已在路上！
时光匆匆，虽未曾谋面，却相遇于斯，实在是莫大的缘分，感谢您的到访 ！

• 题目：
  给定一个整数数组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
• 代码：

class Solution:
    def canThreePartsEqualSum(self, A: List[int]) -> bool:
        sum_val = sum(A)
        if sum_val % 3 != 0:
            return(False)
        temp = sum_val // 3
        i,j = 0,len(A) - 1
        left_val,right_val = 0,0
        while i < j:
            if left_val != temp:
                left_val += A[i]
                i += 1
            if right_val != temp:
                right_val += A[j]
                j -= 1
            if left_val == temp and right_val == temp:
                return(True)
        return(False)
# 执行用时 :448 ms, 在所有 Python3 提交中击败了65.72%的用户
# 内存消耗 :20.6 MB, 在所有 Python3 提交中击败了5.17%的用户

• 算法说明：
  先判断A的所有元素之和能否被3整除，不能整除，直接返回False；如果能整除，使用双指针从左右两端向中间遍历,分别求出两端的累加和，条件是中间的一段不能为空，即循环条件中要满足 i < j 当两端的累加和分别为总和的三分之一时，返回True，否则，循环结束返回False。