给你一个整数数组 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]) 就可以将数组三等分。
示例 1:
输出:[0,2,1,-6,6,-7,9,1,2,0,1]
输出:true
解释:0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
示例 2:
输入:[0,2,1,-6,6,7,9,-1,2,0,1]
输出:false
示例 3:
输入:[3,3,6,5,-2,2,5,1,-9,4]
输出:true
解释:3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4
思路:我刚开始的思路是先用sum计算累加和,如果能被3除尽 则除3然后继续,否则返回false。接下来用cur_sum记录每次累加一部分的和,当等于sum的时候,cnt加1。最后退出循环后判断,当cnt正好是3,并且cur_sum中没有别的数时,返回true,否则返回false。
bool canThreePartsEqualSum(vector<int>& A) {
int sum = 0, cur_sum = 0, cnt = 0;
for (int i = 0; i < A.size(); i++) sum += A[i];
if (sum % 3 == 0) sum /= 3;
else return false;
for (int i = 0; i < A.size(); i++)
{
cur_sum += A[i];
if (cur_sum == sum)
{
++cnt;
cur_sum = 0;
}
}
if (cnt == 3 && cur_sum == 0) return true;
return false;
}
但是: 这个做法有一个样例无法通过:
输入:[10,-10,10,-10,10,-10,10,-10]
输出:false
预期:true
可以看出这个算法有一个缺陷,没考虑到sum=0的情况,正如这个样例所示,最后cnt算出来由4个,但其实可以[10,-10,10,-10] [10,-10] [10,-10]这样划分。所以最后的判断cnt应该大于等于3.
bool canThreePartsEqualSum(vector<int>& A) {
int sum = 0, cur_sum = 0, cnt = 0;
for (int i = 0; i < A.size(); i++) sum += A[i];
if (sum % 3 == 0) sum /= 3;
else return false;
for (int i = 0; i < A.size(); i++)
{
cur_sum += A[i];
if (cur_sum == sum)
{
++cnt;
cur_sum = 0;
}
}
return cnt >= 3;
}
2.双指针法
思路:参考了题解区一位老哥的做法。两边开始遍历,当两边是sum的时候,则中间没遍历到的一定也是sum。注意while循环条件,lef和right中间必须至少有一个,避免[1, -1, 1, -1]的情况发生。
accumulate详解
bool canThreePartsEqualSum(vector<int>& A) {
int left = 0, right = A.size() - 1;
int left_sum = A[left], right_sum = A[right];
int sum = accumulate(A.begin(), A.end(), 0);
if (sum % 3 != 0) return false;
sum /= 3;
while (right - left > 1) //right和left之间至少有1个
{
if (left_sum == sum && right_sum == sum) return true;
if (left_sum != sum) left_sum += A[++left];
if (right_sum != sum) right_sum += A[--right];
}
return false;
}
注意写成这样会有样例无法通过:
class Solution {
public:
bool canThreePartsEqualSum(vector<int>& A) {
int left = 0, right = A.size() - 1;
int left_sum = 0, right_sum = 0; //区别
int sum = accumulate(A.begin(), A.end(), 0);
if (sum % 3 != 0) return false;
else sum /= 3;
while (right - left > 1)
{
if (left_sum == sum && right_sum == sum) return true;
if (left_sum != sum) left_sum += A[left++]; //区别
if (right_sum != sum) right_sum += A[right--]; //区别
}
return false;
}
};
因为是后置递增,所以先累加之后再自增,而自增之后可能就无法通过while循环了,
比如样例:
输入:[3,3,6,5,-2,2,5,1,-9,4]
输出:false
预期:true
在right走到A[3]的5然后累加之后,right直接走到A[2]了,这时将不执行while循环内代码,所以不会返回true。直接返回false。
而前置递增先加之后直接递增,避免了这种情况。