（Python）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

Python实现

思路 循环一遍A数据，遇到等于总值1/3的时候，记录坐标值，最终判断三部分是否均满足和为总值的1/3
特别注意最后一个测试用例：[1,-1,1,-1]

class Solution(object):
    def canThreePartsEqualSum(self, A):
        """
        :type A: List[int]
        :rtype: bool
        """
        s = sum(A)/3
        a = 0  #记录每部分的求和
        c = 0  #记录第几部分
        r=[]   #记录前两部分的截止坐标
        flag = False   #判断第三部分是否为空
        for i in range(len(A)): 
            flag = True 
            a = a+A[i]  
            if a == s and c<2:  #判断a等于总值的1/3且这是第一/二部分
                flag = False   #判断不是第三部分
                r.append(i)   #将此时下标存入r中
                a = 0     #a值清空
                c = c + 1   #记录下面进入第c+1部分计算
        if a == s and r[0]<r[1] and flag:#如果第三部分a等于总值的1/3且记录第一截止点小于第二截止点且第三部分非空
            return True
        else:
            return False

【甜点来袭1】list的查询：①按坐标查询列表值：list[坐标号]②按列表值查询坐标号：list.append(列表值)