leetcode 1013. 将数组分成和相等的三个部分(python)

最新推荐文章于 2022-06-11 21:25:51 发布

洛洛洛洛洛洛洛

最新推荐文章于 2022-06-11 21:25:51 发布

阅读量808

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/weixin_44740082/article/details/90202788

leetcode 专栏收录该内容

242 篇文章

订阅专栏

该博客主要介绍LeetCode的一道题目，即如何将给定的整数数组分成和相等的三个非空部分。通过举例说明了问题的要求，并提出了解决方案：首先检查数组总和是否能被3整除，然后使用双向指针方法尝试从数组两端找到满足三等分条件的元素。如果找到则返回true，否则返回false。

题目链接

题目描述：

给定一个整数数组 A，只有我们可以将其划分为三个和相等的非空部分时才返回 true，否则返回 false。

形式上，如果我们可以找出索引 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]) 就可以将数组三等分。

示例 1：

输出：[0,2,1,-6,6,-7,9,1,2,0,1]
输出：true
解释：0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
示例 2：

输入：[0,2,1,-6,6,7,9,-1,2,0,1]
输出：false
示例 3：

输入：[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

解题思路：

先计算数组的和看是否能被3整除，如果不能直接返回False，如果能用双向指针，从两端同时向中间找等于3等分的数值，如果能找到，说明可以三等分

class Solution(object):
    def canThreePartsEqualSum(self, A):
        """
        :type A: List[int]
        :rtype: bool
        """
        if sum(A)%3!=0:return False
        aver=sum(A)//3
        begin,end=0,len(A)-1
        leftsum,rightsum=0,0
        while begin<end:
            if leftsum!=aver:
                leftsum+=A[begin]
                begin+=1
            if rightsum!=aver:
                rightsum+=A[end]
                end-=1
            if leftsum==aver and rightsum==aver:
                return True
        return False

在这里插入图片描述