将数组分成和相等的三个部分【leetcode每日一题】

最新推荐文章于 2023-02-09 18:31:44 发布

原创最新推荐文章于 2023-02-09 18:31:44 发布 · 792 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #数据结构

Leetcode 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种简单直观的算法，用于判断一个整数数组是否可以被划分为三个连续子数组，每个子数组的元素之和相等。通过求和、遍历和计数，该算法能有效解决这一问题，适合算法初学者学习。

在这里插入图片描述
思路：
1.首先数组里面的数小于三个，直接返回false；
2.对数组里面的数求和，和不是3的整数倍，直接返回false；
3.从左往右遍历，如果s为和的1/3，就让cnt++，s=0；

在第三步考虑到如果你求出来三个区间满足但是最后还有剩余怎么办？
没关系，我们接着加，最后只要判断cnt==3并且剩余的几个数的和s等于0就可以了。

这个思路非常直接，也不用考虑什么细节，适合新手学习，如果有问题在下方留言，欢迎交流！

class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        if(A.size()<3)  return false;
        int sum=accumulate(A.begin(),A.end(),0);
        if(sum%3)   return false;
        int cnt=0,s=0;
        for(int i=0;i<A.size();i++)
        {
            s+=A[i];
            if(s==sum/3&&cnt<3)    {cnt++;s=0;}
        }    
        if(cnt==3&&!s)  return true;
        return false;
    }
};