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

地址：

力扣https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum/

题目：

给你一个整数数组 arr，只有可以将其划分为三个和相等的 非空 部分时才返回 true，否则返回 false。

形式上，如果可以找出索引 i + 1 < j 且满足 (arr[0] + arr[1] + ... + arr[i] == arr[i + 1] + arr[i + 2] + ... + arr[j - 1] == arr[j] + arr[j + 1] + ... + arr[arr.length - 1]) 就可以将数组三等分。

示例 1：

输入：arr = [0,2,1,-6,6,-7,9,1,2,0,1] 输出：true 解释：0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1

示例 2：

输入：arr = [0,2,1,-6,6,7,9,-1,2,0,1] 输出：false

示例 3：

输入：arr = [3,3,6,5,-2,2,5,1,-9,4] 输出：true 解释：3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4

提示：

3 <= arr.length <= 5 * 104 -104 <= arr[i] <= 104

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

既然要分为三组和，那么首先要求和，每一部分和都是其 1/3

如果总和不能被 3 整除，直接false

用双指针分别从数组[0]，数组[len-1]向中间靠拢，分别求左值和，右值和

中间和 = 总和 - 左值和 - 右值和

这里的坑就是：

1. 不能发现 左值和 = 右值和，就跳出循环，这个时候的左值还未达到 1/3，自然需要继续累积

2. 不能发现 左值和 = 右值和，并且等于 1/3 ，跳出循环求得中间和也相等，存在一种可能就是

        总和 = 0，左值和 = 右值和 = 0，比如 [1,-1, -1, 1]

   所以还要看中间数的个数是否为 0，只有不是的情况才能继续判定

这个思路计算比较多，而且考虑边界比较多，测试用例出错几率较大

另外一种思路比较简单，参考方法二

方法一、双指针求和

bool canThreePartsEqualSum(int* arr, int arrSize){
    int goal = 0;
	int lsum = 0, msum = 0, rsum = 0;
	int sums = 0;
    int cnt = arrSize;
	int i,j;
	
	for(i=0; i<arrSize; i++)
		sums += arr[i];
	
    if(sums % 3 != 0)
        return false;

    goal = sums/3;

    i = 0;
    j = arrSize - 1;
    lsum += arr[i];
    rsum += arr[arrSize-1];
    cnt-=2;

    while( cnt > 0)
    {
        if(lsum != goal)
        {
            i++;
            lsum += arr[i];
            cnt--;
        }
        
        if(rsum != goal)
        {
            j--;
            rsum += arr[j];
            cnt--;
        }

        if(lsum == goal && lsum == rsum)
            break;
        
    }
    
    msum = sums - lsum - rsum;
    
    if(cnt > 0 && lsum == msum && rsum == msum)
		return true;
    else
    	return false;    
}

方法二、求倍数的个数

我们知道 1/3 的和，那么至少存在 3 个这样的和，也有可能有多，比如和为0 或者 3n 个这样的组合

所以我们只需要两个步骤即可完成

1. 统计总和

2. 再求 1/3 的和，统计个数++，如果统计个数最后少于 3 ，那就是false

bool canThreePartsEqualSum(int* arr, int arrSize){
    int sum = 0;
    int tmp = 0;
    int goal = 0;
    int cnt = 0;
    int i;

    for(i=0; i<arrSize; i++)
        sum += arr[i];
    
    if(sum % 3 != 0)
        return false;
    
    goal = sum / 3;

    for(i=0; i<arrSize; i++)
    {
        tmp += arr[i];
        if(tmp == goal)
        {
            cnt++;
            tmp = 0;
        }
    }

    if(cnt < 3)
        return false;
    
    return true;
}

查看更多刷题笔记