- Notifications
You must be signed in to change notification settings - Fork 46.7k
/
Copy pathfind_max.py
84 lines (75 loc) · 2.48 KB
/
find_max.py
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
from __future__ importannotations
deffind_max_iterative(nums: list[int|float]) ->int|float:
"""
>>> for nums in ([3, 2, 1], [-3, -2, -1], [3, -3, 0], [3.0, 3.1, 2.9]):
... find_max_iterative(nums) == max(nums)
True
True
True
True
>>> find_max_iterative([2, 4, 9, 7, 19, 94, 5])
94
>>> find_max_iterative([])
Traceback (most recent call last):
...
ValueError: find_max_iterative() arg is an empty sequence
"""
iflen(nums) ==0:
raiseValueError("find_max_iterative() arg is an empty sequence")
max_num=nums[0]
forxinnums:
ifx>max_num: # noqa: PLR1730
max_num=x
returnmax_num
# Divide and Conquer algorithm
deffind_max_recursive(nums: list[int|float], left: int, right: int) ->int|float:
"""
find max value in list
:param nums: contains elements
:param left: index of first element
:param right: index of last element
:return: max in nums
>>> for nums in ([3, 2, 1], [-3, -2, -1], [3, -3, 0], [3.0, 3.1, 2.9]):
... find_max_recursive(nums, 0, len(nums) - 1) == max(nums)
True
True
True
True
>>> nums = [1, 3, 5, 7, 9, 2, 4, 6, 8, 10]
>>> find_max_recursive(nums, 0, len(nums) - 1) == max(nums)
True
>>> find_max_recursive([], 0, 0)
Traceback (most recent call last):
...
ValueError: find_max_recursive() arg is an empty sequence
>>> find_max_recursive(nums, 0, len(nums)) == max(nums)
Traceback (most recent call last):
...
IndexError: list index out of range
>>> find_max_recursive(nums, -len(nums), -1) == max(nums)
True
>>> find_max_recursive(nums, -len(nums) - 1, -1) == max(nums)
Traceback (most recent call last):
...
IndexError: list index out of range
"""
iflen(nums) ==0:
raiseValueError("find_max_recursive() arg is an empty sequence")
if (
left>=len(nums)
orleft<-len(nums)
orright>=len(nums)
orright<-len(nums)
):
raiseIndexError("list index out of range")
ifleft==right:
returnnums[left]
mid= (left+right) >>1# the middle
left_max=find_max_recursive(nums, left, mid) # find max in range[left, mid]
right_max=find_max_recursive(
nums, mid+1, right
) # find max in range[mid + 1, right]
returnleft_maxifleft_max>=right_maxelseright_max
if__name__=="__main__":
importdoctest
doctest.testmod(verbose=True)