leetcode每日一题 -将数组分为和相等的三个部分

题目分析

• 和不被3整除的一定不行
• 和可以被3整除的，有两个数组相等=sum/3,并且数组元素还有剩余即可
• 其余均不符合

Java代码

public boolean canThreePartsEqualSum(int[] A) {
        int sum = 0;
        for(int i=0;i<A.length;i++){
            sum += A[i];
        }
        int count = 0;
        if(sum%3 != 0){
            return false;
        }else{
            sum = sum/3;
            int curSum = 0;
            for(int i=0;i<A.length;i++){
                curSum += A[i];
                if(curSum == sum){
                    curSum = 0;
                    count++;
                    if(count == 2 && i!=A.length-1){
                        return true;
                    }
                    continue;
                }
            }
        }
        return false;
    }