LeeCode每日一题--将数组分成和相等的三个部分

  【前言】坚持日更LeeCode刷题系列

    不积跬步,无以至千里;不积小流,无以成江海。愿与诸君共勉!


  【题目】1013.将数组分成和相等的三个部分

    题目描述:给你一个整数数组 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
        -10^4 <= A[i] <= 10^4


    思路一:1.判断总和能否被3整除,可以的话继续下一步
          2.找出前段和等于sum/3时候的边界index
          3.找出后段和等于sum/3时候的边界index(思考下找中间段代替)
          4.通过前面的两个index,可以得到中间部分,并计算出它的总和
          5.判断中间部分的总和与sum/3是否相等,相等返回true,否则返回false
          具体代码如下:

class Solution(object):
    def canThreePartsEqualSum(self, A):
        """
        :type A: List[int]
        :rtype: bool
        """
        num_sum = sum(A)
        if(num_sum//3 != num_sum/3): #判断总和能否被等分为三份
            return False
        else:
            flag1 = 0  #用于标记第一个部分的边界
            for i in range(len(A)):    #找出前段和等于sum/3的边界值
                flag1 = flag1+A[i]
                if flag1 == num_sum//3:
                    flag1 = i
                    break
            flag2 = 0
            for i in range(len(A)-1,flag1,-1):  #找出后段和等于sum/3的边界值
                flag2 = flag2+A[i]
                if flag2 == num_sum//3:
                    flag2 = i
                    break
            if flag1+1==flag2:  #思考下如果不加这个判断的话,可以吗
                return False
            else:
                num_sum2 = sum(A[flag1+1:flag2])  #算出中间部分的值
                if(num_sum2 == num_sum/3):  #判断中间部分的值是否等于sum/3
                    return True
                else:
                    return False 

    运行结果:
    在这里插入图片描述


    思路二:另外官方给出了另一种解题思路,即找出能否在数组中找出两个切分点,如果能找到两个切分点,那么就证明该数组能被三等分。下面会附上链接,有兴趣的朋友可以自行去了解。


    关于其中一些知识的链接:
    算法链接


    分享就到这里了,欢迎大家一起交流讨论。


    注明

    题目来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum

发布了32 篇原创文章 · 获赞 62 · 访问量 1297

猜你喜欢

转载自blog.csdn.net/Mingw_/article/details/104795507
今日推荐