【前言】坚持日更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