将数组分成和相等的三个部分（partition-array-into-three-parts-with-equal-sum）

将数组分成和相等的三个部分（partition-array-into-three-parts-with-equal-sum）

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

提示：
1.3 <= A.length <= 50000
2.-10^4 <= A[i] <= 10^4

思考

题目这里说的三段是说不打乱顺序的。

循环的方法

package question1013;

class Solution {
	
    public boolean canThreePartsEqualSum(int[] A) {
		int len = A.length;
		int[] arr = A;
		for (int i = 1; i < len - 1; i++) {
			for (int j = i + 1; j < len;) {
				int a1 = sumarr(0, i, arr);
				int a2 = sumarr(i, j, arr);
				if (a1 == a2) {
					if (a2 == sumarr(j, len, arr)) {
						return true;
					} else {
						j++;
					}
				} else {
					j++;
				}
			}
		}
		return false;
    }
    
    private int sumarr(int a, int b,int[] arr) {
		int sum = 0;
		for (int i = a; i < b; i++) {
			sum += arr[i];
		}
		return sum;
	}

	public static void main(String[] args) {
		int[] a = {0,2,1,-6,6,-7,9,1,2,0,1};
		int[] b = {0,2,1,-6,6,7,9,-1,2,0,1};
		int[] c = {3,3,6,5,-2,2,5,1,-9,4};
		Solution s = new Solution();
		
		System.out.println(s.canThreePartsEqualSum(a)==true);
		System.out.println(s.canThreePartsEqualSum(b)==false);
		System.out.println(s.canThreePartsEqualSum(c)==true);

	}
    
}

测试示例没有问题，但是提交后，超时了。这里需要一点技巧。一个数组可以被分为和相等的三倍分，也就是说
a1+a2+a3=sum，我们知道数组全部元素的和sum可以直接求出，那数组a1的和可以直接求出来吗，答案是肯定的。
因为a1=a2=a3，所以3x a1 = sum所以，单个数组的和是sum/3。
因为这里的数组中的元素有正也有负，所以不一定都是3x a1 = sum也有可能是4x a1 = sum，5x a1 = sum。

大神的代码

代码一

    public boolean canThreePartsEqualSum(int[] A) {
        int sum = 0;
        for (int num: A) {
            sum += num;
        }
        // 数组A的和如果不能被3整除直接返回false
        if (sum % 3 != 0) {
            return false;
        }
        // 遍历数组累加，每累加到目标值cnt加1，表示又找到1段
        sum /= 3;
        int curSum = 0, cnt = 0;
        for (int i = 0; i < A.length; i++) {
            curSum += A[i];
            if (curSum == sum) {
                cnt++;
                curSum = 0;
            }
        }
        // 最后判断是否找到了3段（注意如果目标值是0的话可以大于3段）
        return cnt == 3 || (cnt > 3 && sum == 0);
    }

要理解为什么cnt > 3。

代码二

找到两段后就可以break了，优化一下。

        int sum = 0;
        for (int num: A) {
            sum += num;
        }
        // 数组A的和如果不能被3整除返回false
        if (sum % 3 != 0) {
            return false;
        }
        // 遍历数组累加，每累加到目标值cnt加1，表示又找到1段,
        // 找到2段后就返回true（i只能到数组A的倒数第二个元素，保证了有第3段）
        sum /= 3;
        int curSum = 0, cnt = 0;
        for (int i = 0; i < A.length - 1; i++) {
            curSum += A[i];
            if (curSum == sum) {
                cnt++;
                if (cnt == 2) {
                    return true;
                }
                curSum = 0;
            }
        }
        return false;

参考资料

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