难度:简单
类型:数组
题目:一个整数数组 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
方法1:
class Solution:
def canThreePartsEqualSum(self, A: List[int]) -> bool:
'''
根据sum值找avg
根据avg确定分界
'''
# 判断avg是否为整数。不能用isinstance(sum(A)/3, int), 因为/计算结果一定有小数位.0
if sum(A) % 3 == 0: avg = sum(A) // 3
else: return False
# 确定分界i, j
for i in range(1, len(A)):
if sum(A[0: i]) == avg:
B = A[i :]
break
if sum(A[0: i]) != avg: return False
for j in range(1, len(B)):
if sum(B[0: j]) == avg and sum(B[j: ]) == avg:
return True
return False
方法2:由于方法1多次使用sum,时间消耗大,改用累加,降低时间复杂度
class Solution:
def canThreePartsEqualSum(self, A: List[int]) -> bool:
if sum(A) % 3 == 0: avg = sum(A) // 3
else: return False
n, i, cur = len(A), 0, 0
while i < n:
cur += A[i]
if cur == avg:
break
i += 1
if cur != avg: return False
j = i + 1
while j + 1 < n:
cur += A[j]
if cur == avg * 2: return True
j += 1
return False
时间复杂度:O(N)
空间复杂度:O(1)
方法 | 提交结果 | 执行用时 | 内存消耗 | 语言 |
---|---|---|---|---|
方法1 | 通过 | 6664 ms | 18.7 MB | Python3 |
方法2 | 通过 | 108 ms | 18.7 MB | Python3 |
题目来源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum