LeetCode 1013. 将数组分成和相等的三个部分 暴力⑧可取

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

乍一看先想到暴力破解j和i

想都不用想要么超时要么效率很感人

毕竟i和j大约有一半length² 的组合，意味着验算这些组合要遍历这么多次数组，想想都慢好吧

class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
        for(int i=1;i<A.length;i++){
            for(int j=i+1;j<A.length;j++){
                if(help(A,i,j)){
                    return true;
                }
            }
        }
        return false;
    }
    public boolean help(int[] A,int I,int J){
        int a1=0,a2=0,a3=0;
        for(int i=0;i<I;i++){
            a1+=A[i];
        }
        for(int i=I;i<J;i++){
            a2+=A[i];
        }
        for(int i=J;i<A.length;i++){
            a3+=A[i];
        }
        return a1==a2&&a1==a3;
    }
}

果不其然

用空间换时间！建一个数组sum，表示对应下标和他之前所有数字的和

那么，三段分别是：sum[i]、sum[j]-sum[i]、sum[length-1]-sum[j]

class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
        for(int i=1;i<A.length;i++){
            A[i]+=A[i-1];
        }
        for(int i=0;i<A.length-2;i++){
            for(int j=i+1;j<A.length-1;j++){
                if(help(A,i,j)){
                    System.out.print(i+"---"+j);
                    return true;
                }
            }
        }
        return false;
    }
    public boolean help(int[] A,int i,int j){
        return A[i]==A[j]-A[i]&&A[i]==A[A.length-1]-A[j];
    }
}

虽然已经优化了实现，但是暴力解法似乎注定超过时间限制（不过这次通过了所有样例，只是总时间超时，有进步？）

似乎已经束手无策了？

要注意到一点：

假设数组总和为sum，那么可以分成相等的三段等价于：

数组可以分成三段，每一段的和都为sum/3!

class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
       int sum=0;
       for(int i=0;i<A.length;i++){
           sum+=A[i];
       }
       //不能被3整除的数组一定没有办法三等分
       if(sum%3!=0){
           return false;
       }
       int count=0;
       int times=0;
       //寻找有几段和为sum/3的子段
       for(int i=0;i<A.length;i++){
           count+=A[i];
           if(count==sum/3){
               times++;
               count=0;
           }
       }
       //有三段或者，sum=0且有3段或者以上，都可以
       return times==3||(sum==0&&times>=3);
    }
}