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

思路

1.分成相等的三分则意味着数组和一定为3的倍数,所以当sum的值不是散的倍数时,则一定为false;
2.使用双指针分别从头尾访问并累加元素值,当leftsum和rightsum都为sum/3时,代表此时数组被平均分为3分,中间那份一定也是sum/3.
3.当循环结束没有找到left和right的合适位置则代表不能
4.方法2和方法1思路相似,体会flag的用法.

代码

方法1

 //将数组分成和相等的三个部分
        public boolean canThreePartsEqualSum(int[] A) {
         //1.分成和相等的三个部分则意味着和一定是3的倍数，如果不是直接返回false
            int sum = 0;
            for (int i = 0;i<A.length;i++){
               sum += A [i];
            }
            if (sum%3 != 0){
                return false;
            }
        //2.使用双指针分别从头尾出发，加快速率
            int left = 0;
            int right = A.length-1;
            int leftsum = A[left];
            int rightsum =A[right];
            //循环将A中元素按左右指针分别求和
            //将left+1<right作为循环条件，若是left<right的，则不能保证一定是将数组分为3部分
            //如：[1，-1,1，-1]
            while (left+1<right){
                if (leftsum != sum/3){
                    //左边不等于1/3，则更新left的值和左边sum的值
                    left++;
                    leftsum += A[left];
                }
                if (rightsum != sum/3){
                    //右边不等于1/3，则更新right的值和右边sum的值
                    right--;
                    rightsum += A[right];
                }
                //此时代表sum已经被分为和相等的三部分（中间部分肯定也为1/3）
                if (rightsum == sum/3&&leftsum == sum/3){
                    return true;
                }
            }
            //循环结束还没找到left和right合适的位置则代表不能平均分
            return false;


        }

方法2:

   //将数组分成和相等的三个部分,方法2
        public boolean canThreePartsEqualSum2(int[] A) {
        int sum = 0;
        for(int i: A){
            sum += i;
        }
        if(sum%3 != 0){
            // 总和不是3的倍数，直接返回false
            return false;
        }
        int s = 0;
        int flag = 0;//记录是前中后哪个部分
        for(int i:A){
            s += i;
            if(s == sum/3){
                //当sum的值累加到sum/3时,代表此时前段已经找到了
                //更新flag,开始记录中段
                flag++;
                s = 0;
            }
        }
        //当flag>3时,一定是后段的后面元素相加为0,所以不用考虑
        // flag不一定等于3，例如[1,-1,1,-1,1,-1,1,-1]
        return flag >= 3;
    }