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

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

给定一个整数数组 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

提示：

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

class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
        /*
        关键点：一个数组所有元素的和可以三等分
        步骤：首先求出数组元素和三等分后的结果partsum
        遍历数组相加各元素直到等于partsum，找到第一部分子数组
        从第一次结尾的位置继续遍历找第二个和等于partsum的子数组
        最后找第三个子数组
        比较三个子数组和是否相等，若相等则能分，不相等则不能分
        时间O（n）空间O（1）
        */
        int sumA = 0;
        for(int t : A){
            sumA += t;
        }
        int partSum = sumA / 3;
        
        int firstpart = 0;
        
        int tsum1 = 0;
        while(firstpart < A.length && tsum1 != partSum){
            tsum1 += A[firstpart];
            firstpart++;
        }
        
        int secondpart = firstpart;
        int tsum2 = 0;
        while(secondpart < A.length && tsum2 != partSum){
            tsum2 += A[secondpart];
            secondpart++;
        }
        
        int thirdpart = secondpart;
        int tsum3 = 0;
        while(thirdpart < A.length){
            tsum3 += A[thirdpart];
            thirdpart++;
        }
        

        if(tsum1 == tsum2 && tsum2 == tsum3){
            return true;
        }else{
            return false;
        }
    }
}