这是小川的第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+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。

以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!

内容来源于网络如有侵权请私信删除
你还没有登录,请先登录注册
  • 还没有人评论,欢迎说说您的想法!