这是小川的第378次更新,第406篇原创
01 看题和准备
今天介绍的是LeetCode算法题中Easy级别的第240题(顺位题号是1013)。给定一个整数数组A,当且仅当我们可以将数组分成具有相等和的三个非空部分时才返回true。
形式上,我们可以对数组进行分区,如果我们能找到索引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]
。
例如:
输入:[0,2,1,-6,6,-7,9,1,2,0,1]
输出:true
说明:0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
输入:[0,2,1,-6,6,7,9,-1,2,0,1]
输出:false
输入:[3,3,6,5,-2,2,5,1,-9,4]
输出:true
说明:3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4
注意:
-
3 <= A.length <= 50000
-
-10000 <= A[i] <= 10000
02 解题
题目的意思是数组中的元素可以分为三部分,每部分的和都相等,如果给的数组符合上述条件,就返回true
。
思路:先把A的总和求出来,判断是否能被3整除,如果可以,就再遍历A数组,将元素累加求和,如果当前累加的元素之和等于总和的三分之一,就记一次数,将存当前累加的元素之和的变量重置为0,继续遍历剩下的元素,最后判断记数次数是否等于3即可。
public boolean canThreePartsEqualSum(int[] A) {
int totalSum = 0, eachSum = 0;
for (int num : A) {
totalSum += num;
}
if (totalSum%3 != 0) {
return false;
}
eachSum = totalSum/3;
int n = A.length, temSum = 0, count = 0;
for (int i = 0; i<n; i++) {
temSum += A[i];
if (temSum == eachSum) {
count++;
temSum = 0;
}
}
return count == 3;
}
03 小结
算法专题目前已连续日更超过七个月,算法题文章246+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。
以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!