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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

只需要求出数组的总和，然后除以3，就可以得到每组的和了
如果不能整除，则不可能三分
然后只需要按顺序寻找从A[0]加到A[i]中满足1倍部分和、2倍部分和的分割点即可

代码

class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        // 累计求和
        for(int i = 1; i < A.size(); i++){
            A[i] += A[i - 1]; 
        //   cout<<A[i]<<endl;
        }
        // 是否可三分
        if(A[A.size() - 1] % 3 != 0){
        //    cout<<"No"<<endl;
            return false;
        }
        // 计算每组的和
        int sum_3 = A[A.size() - 1] / 3;
        int i = 0; // 寻找第一个切分点
        for(; i < A.size(); i++){
            if(A[i] == sum_3){
                break;
            }
        }// 寻找第二个切分点
        for(i++; i < A.size() - 1; i++){
            if(A[i] == 2 * sum_3){
            //    cout<<i;
                return true;
            }
        }
        return false;
    }
};