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

示例 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

提示：

3 <= A.length <= 50000
-10^4 <= A[i] <= 10^4

暴力超时算法

优先想到的就是暴力算法，但是超时了，难顶，我去看了平台测试的 那个数据，，，，，，长的就离谱，，，，，超时了，没办法，优化呗，就发现重复计算了sumJ，就用记忆数组，但还是超时，难顶，只有去看题解了

class Solution {

	public static boolean canThreePartsEqualSum(int[] A) {
		int sumI=0;
		//构造一个记忆数组
		int[] rec=new int[A.length];
		rec[A.length-1]=A[A.length-1];
		for(int j=A.length-2;j>=0;j--)
			rec[j]=rec[j+1]+A[j];
		for(int i=0;i<A.length;i++) {
			sumI+=A[i];
			int sumJ=0;
			for(int j=A.length-1;j>i;j--) {
				sumJ=rec[j];
				if(sumJ==sumI&&i!=(j-1)) {
					int sum=0;
					for(int k=i+1;k<j;k++)
						sum+=A[k];
					if(sum==sumI)
						{System.out.println(i+"-"+j);
						return true;}
				}
			}
		}
		return false;
    }
}

解题思路

参考了解题解
根据题目要求可以看出，数组如果能够三等分，就必须满足三部分的和都是总和的三分之一。
所以先计算总和，然后判断和是否是3的整数倍，如果不是，直接返回false；
倘若是三的整数倍，就计算前部分和后部分的和（用指针i和j,要满足i < j 且i和j之间不能没有元素），如果满足两部分和都等于总和的三分之一，就return true；

代码

class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
        int sum=0;
		for(int a:A)
			sum+=a;
		if(sum%3!=0)
			return false;
		for(int i=0,j=A.length-1,sumI=A[0],sumJ=A[A.length-1];i<j&&i!=j-1;) {
			if(sumI==sum/3&&sumJ==sum/3)
				{//System.out.println(i+"***"+j);
				return true;}
			if(sumI!=sum/3)
			{
				sumI+=A[++i];
			}
			if(sumJ!=sum/3)
			{
				sumJ+=A[--j];
			}
		}
		return false;
    }
}