给你一个整数数组 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]) 就可以将数组三等分。
直接法
-
首先计算数组 AA 中所有数字总和 sum
-
遍历数组 AA 查找和为 sum / 3 的子数组个数
-
如果找到了三个这样的子数组则返回 true
-
找不到三个就返回 false
class Solution {
public:
bool canThreePartsEqualSum(vector<int>& A) {
int sum = accumulate(A.begin(), A.end(), 0);
if (sum % 3 != 0) {
return false;
}
int count = 0, subSum = 0;
for (int i = 0; i < A.size(); i ++) {
subSum += A[i];
if (subSum == sum / 3) {
count ++;
subSum = 0;
}
if (count == 3) {
return true;
}
}
return false;
}
};
双指针法
如果能等分,数组的所有元素和肯定能被3整除。所以在所有元素和能被3整除这个前提下,我们从数组左右两端分别找到使左右两边的子数组都等于等分和的索引位置。此时如果左右索引中间还有剩余元素,说明中间剩余元素的和也是等分和。
class Solution {
public:
bool canThreePartsEqualSum(vector<int>& A) {
int sum = accumulate(A.begin(), A.end(), 0);
if (sum % 3 != 0) {
return false;
}
int left = 0;
int leftsum = A[left];
int right = A.size()-1;
int rightsum = A[right];
while(left+1<right)
{
if(leftsum == sum / 3 && rightsum == sum / 3)
{
return true;
}
if(leftsum != sum / 3)
{
leftsum += A[++left];
}
if(rightsum != sum / 3)
{
rightsum += A[--right];
}
}
return false;
}
};