【Leetcode】1013. Partition Array Into Three Parts With Equal Sum（配数学证明）

题目地址：

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

给定一个数组$A$，问是否能将$A$拆成$3$段，使得每段非空，并且每段数字和相等。

先求一下$A$的数字和$s$，则每段的数字和应当为$s/3$。如果$s$不能被$3$整除那直接返回false。接下来开始截取$A$中和为$s/3$的一段，并对段数进行计数，设最终计数为$c$。如果$c=3$或者$s=0$并且$c\ge 3$，就说明可以拆成和相等的$3$段，返回true，否则返回false。

算法正确性证明：
首先算法返回true，则$A$一定能拆成和相等的$3$段，这一点很显然。如果算法返回false，但$A$仍然可以拆成和相等的$3$段，如果$s=0$，那么算法得到的$c\ge 3$，应该返回了true，矛盾；如果$s\ne 0$，首先算法算出的$c$一定满足$c\ge 3$，如果$c>3$，那么说明存在超过$3$段的子段和等于$s/3$，则$A$总和应该等于$cs/3\ne s$矛盾，所以$c=3$，应该返回true，矛盾。所以算法正确。

代码如下：

public class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
        int sum = 0;
        for (int i = 0; i < A.length; i++) {
            sum += A[i];
        }
        
        if (sum % 3 != 0) {
            return false;
        }
    
        int curSum = 0, count = 0;
        for (int i = 0; i < A.length; i++) {
            curSum += A[i];
            if (curSum == sum / 3) {
                count++;
                curSum = 0;
            }
        }
        
        return count == 3 || (sum == 0 && count >= 3);
    }
}

时间复杂度$O(n)$，空间$O(1)$。