forked from yubinbai/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
42 lines (37 loc) · 1.12 KB
/
main.py
File metadata and controls
42 lines (37 loc) · 1.12 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class Solution:
# @param A, a list of integers
# @return an integer
def maxSubArray(self, A):
# kadane's dynamic programming algorithm
maxSum = -1 * (1 << 30)
currSum = maxSum
for a in A:
currSum = max(a, a + currSum)
maxSum = max(maxSum, currSum)
return maxSum
def kadane(array):
# kadane's dynamic programming algorithm
maxL, maxR, maxSum = -1, -1, 0.0
currL, currR, currSum = 0, 0, 0.0
for i, a in enumerate(array):
if currSum < 0:
currL, currR, currSum = i, i + 1, a
else:
currR, currSum = i + 1, currSum + a
if maxSum < currSum:
maxL, maxR, maxSum = currL, currR, currSum
return maxL, maxR, maxSum
def kadane2(array):
# kadane's dynamic programming algorithm
maxSum = 0.0
currSum = 0.0
for a in array:
currSum = max(a, a + currSum)
maxSum = max(maxSum, currSum)
return maxSum
if __name__ == '__main__':
s = Solution()
print s.maxSubArray([-2,1,-3,4,-1,2,1,-5,4])
print s.maxSubArray([-1])
print kadane([-1])
print kadane2([-1])