码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode1020

时间:2019-03-24 13:52:10      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:lse   equals   list   leetcode   break   get   ant   span   17.   

 1 class Solution(object):
 2     def canThreePartsEqualSum(self, A: List[int]) -> bool:
 3         n = len(A)
 4         sums = sum(A)
 5         if sums % 3 != 0:
 6             return False
 7         target = sums // 3
 8         
 9         leftpart = 0
10         leftboundary = n -1
11         for i in range(n-2):
12             leftpart += A[i]
13             if leftpart == target:
14                 leftboundary = i
15                 break
16                 
17         rightpart = 0
18         rightboundary = 0
19         for j in range(n-1,1,-1):
20             rightpart += A[j]
21             if rightpart == target:
22                 rightboundary = j
23                 break
24         
25         if leftboundary < rightboundary:
26             return True
27         return False

先判断数组之和是否是3的倍数,如果不是,那么不能三等分。

如果可以,先确定1/3的值,保存在变量target中。

从左向右逐项相加,找到第一组等于target的值的索引,记为i。

从右向左逐项相加,找到第一组等于target的值的索引,记为j。

如果i<j,则可以三等分。

时间复杂度O(n)(三次遍历),64ms,17.4MB。

leetcode1020

标签:lse   equals   list   leetcode   break   get   ant   span   17.   

原文地址:https://www.cnblogs.com/asenyang/p/10587831.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!