-
Notifications
You must be signed in to change notification settings - Fork 29
/
0881-Boats-to-Save-People.py
66 lines (51 loc) · 1.71 KB
/
0881-Boats-to-Save-People.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
'''
You are given an array people where people[i] is the weight of the ith person, and an infinite number of boats where each boat can carry a maximum weight of limit.
Each boat carries at most two people at the same time, provided the sum of the weight of those people is at most limit.
Return the minimum number of boats to carry every given person.
Example 1:
Input: people = [1,2], limit = 3
Output: 1
Explanation: 1 boat (1, 2)
Example 2:
Input: people = [3,2,2,1], limit = 3
Output: 3
Explanation: 3 boats (1, 2), (2) and (3)
Example 3:
Input: people = [3,5,3,4], limit = 5
Output: 4
Explanation: 4 boats (3), (3), (4), (5)
Constraints:
1 <= people.length <= 5 * 104
1 <= people[i] <= limit <= 3 * 104
'''
from typing import List
# Two Pointers Method
class Solution:
def numRescueBoats(self, people: List[int], limit: int) -> int:
people.sort()
left, right, boat = 0, len(people) - 1, 0
while left <= right:
if left == right:
boat += 1
break
if people[left] + people[right] <= limit:
left += 1
boat += 1
right -= 1
return boat
# Negligibly Better Solution, same method
class Solution:
def numRescueBoats(self, people: List[int], limit: int) -> int:
people.sort()
left, right, boat = 0, len(people) - 1, 0
while left <= right:
boat += 1
if left == right:
break
if people[left] + people[right] <= limit:
left += 1
right -= 1
return boat
s = Solution()
answer = s.numRescueBoats([2, 1, 3, 4], 4) # 3
print(answer)