LeetCode 1013. Partition Array Into Three Parts With Equal Sum

Given an array of integers arr, return true if we can partition the array into three non-empty parts with equal sums.

Formally, we can partition the array if we can find indexes i + 1 < j with (arr[0] + arr[1] + ... + arr[i] == arr[i + 1] + arr[i + 2] + ... + arr[j - 1] == arr[j] + arr[j + 1] + ... + arr[arr.length - 1])

Example 1:

Input: arr = [0,2,1,-6,6,-7,9,1,2,0,1]
Output: true
Explanation: 0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1

Example 2:

Input: arr = [0,2,1,-6,6,7,9,-1,2,0,1]
Output: false

Example 3:

Input: arr = [3,3,6,5,-2,2,5,1,-9,4]
Output: true
Explanation: 3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4

Constraints:

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

---

思路挺直观的，先计算平均数，如果不能整除3就是false。然后从左到右算sum，如果等于平均数的话就开始计算下一部分的sum。本来想着接着计算第三部分，结果没考虑到如果第二部分已经到结尾了，并且正好平均数是0，就fail了。fail case：[1,-1,1,-1]。然后看了解答才意识到其实不需要算第三部分，因为如果前两部分都是平均数的话第三部分必然也是平均数，所以可在这时候就判断它是否还没到最后一个，就cover了上面这种corner case。

class Solution {
    public boolean canThreePartsEqualSum(int[] arr) {
        int sum = addSum(arr);
        if (sum % 3 != 0) {
            return false;
        }
        int partialSum = sum / 3;
        int sum1 = 0;
        for (int i = 0; i < arr.length; i++) {
            sum1 += arr[i];
            if (sum1 == partialSum) {
                int sum2 = 0;
                for (int j = i + 1; j < arr.length; j++) {
                    sum2 += arr[j];
                    if (sum2 == partialSum && j != arr.length - 1) {
                        return true;
                    }
                }
            }
        }
        return false;
    }

    private int addSum(int[] arr) {
        int result = 0;
        for (int i : arr) {
            result += i;
        }
        return result;
    }
}

然后看了解答，还有一种2 pointers的思想。左右两边的pointer同时移动，如果它俩还没达到平均数，就接着移，直到两个相遇就说明不成功，或者两个都等于平均数，并且left <= right（注意这里的且，因为有可能正好前半部分和后半部分分别等于平均值，在if里就各自又往前/后移了一步，这时候也还在while循环里满足条件，所以需要排除这种情况），就成功。还有一点很坑的就是还是上面那个corner case，刚开始是两个pointer从最头和最尾走，但是这种情况根本不会进里面的if，pointer根本不会移动，所以就从以前后为基准开始。

class Solution {
    public boolean canThreePartsEqualSum(int[] arr) {
        int sum = addSum(arr);
        if (sum % 3 != 0) {
            return false;
        }
        int partialSum = sum / 3;
        int left = 1;
        int right = arr.length - 2;
        int sum1 = arr[0];
        int sum2 = arr[arr.length - 1];
        while (left < right) {
            if (sum1 != partialSum) {
                sum1 += arr[left];
                left++;
            }
            if (sum2 != partialSum) {
                sum2 += arr[right];
                right--;
            }
            if (sum1 == partialSum && sum2 == partialSum && left <= right) {
                return true;
            }
        }
        return false;
    }

    private int addSum(int[] arr) {
        int result = 0;
        for (int i : arr) {
            result += i;
        }
        return result;
    }
}