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;
}
}