一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第18天，点击查看活动详情。
题目描述
给你一个整数数组 arr，只有可以将其划分为三个和相等的 非空 部分时才返回 true，否则返回 false。 形式上，如果可以找出索引 i + 1 < j 且满足 (arr[0] + arr[1] + ... + arr[i] == arr[i + 1] + arr[i + 2] + ... + arr[j - 1] == arr[j] + arr[j + 1] + ... + arr[arr.length - 1]) 就可以将数组三等分。（出自力扣）
- 示例1
输入：arr = [0,2,1,-6,6,-7,9,1,2,0,1]
输出：true
解释：0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
- 示例2
输入： arr = [0,2,1,-6,6,7,9,-1,2,0,1]
输出： false
- 示例3
输入：arr = [3,3,6,5,-2,2,5,1,-9,4]
输出：true
解释：3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4
提示：
3 <= arr.length <= 5 * 104-104 <= arr[i] <= 104
思路分析
根据题意可知，我们需要将一个数组分成三部分，并且，这三部分中每部分的总和是总数组之和的三分之一，三部分元素个数不需要相同。
- 首先，将整个数组累加，计算出总和
- 若是整个数组的总和不能被3整除，那么这个数组就不能平分成三分，可以直接return false
- 循环整个数组，在每次的循环中，将遍历项相加，再把相加之后的和与之前的总和的三分之一作对比，若是不相等，则继续累加；若是相等，则累加之和清零，符合的次数time加1；若是累加总和num大于数组总和的三分之一，那么可以直接return false；直至循环结束，若是符合的次数time大于等于3，那么证明可以分成总和相等的三个数组。
AC代码
let arr = [0,2,1,-6,6,-7,9,1,2,0,1]
function solution(arr) {
let sum = arr.reduce((a, b) => {return a + b});
if(sum % 3 != 0) {
return false;
}
let mean = sum / 3;
let num = 0;
let time = 0;
for(let i=0; i<arr.length; i++) {
num += arr[i];
if(num === mean) {
num = 0;
time++
}else if(num > mean) {
return false;
}
}
return time >= 3;
}
solution(arr)
