LeetCode.1013-分割数组为三个和相同的部分

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

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