forked from someshgkale123/LeetCode-Easy
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Maximize Distance to Closest Person
56 lines (46 loc) · 1.72 KB
/
Maximize Distance to Closest Person
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
Problem Statement:
In a row of seats, 1 represents a person sitting in that seat, and 0 represents that the seat is empty.
here is at least one empty seat, and at least one person sitting.
Alex wants to sit in the seat such that the distance between him and the closest person to him is maximized.
Return that maximum distance to closest person.
Example 1:
Input: [1,0,0,0,1,0,1]
Output: 2
Explanation:
If Alex sits in the second open seat (seats[2]), then the closest person has distance 2.
If Alex sits in any other open seat, the closest person has distance 1.
Thus, the maximum distance to the closest person is 2.
Example 2:
Input: [1,0,0,0]
Output: 3
Explanation:
If Alex sits in the last seat, the closest person is 3 seats away.
This is the maximum distance possible, so the answer is 3.
Solution:
class Solution:
def maxDistToClosest(self, seats: List[int]) -> int:
currmax=0
prevind=None
for i,j in enumerate(seats):
if j==1:
if prevind==None:
currmax=i
else:
currmax=max(currmax,(i-prevind)//2)
prevind=i
currmax=max(currmax,i-prevind)
return currmax
"""
if seats.count(1)==1:
ind=seats.index(1)
return max(ind-0,len(seats)-1-ind)
track1s=[i+1 for i in range(len(seats)) if seats[i]==1]
maxdist=0
if seats[0]==0:
maxdist=max(maxdist,track1s[0]-1)
for i in range(len(track1s)-1):
maxdist=max(maxdist,math.floor((track1s[i+1]-track1s[i])/2))
if seats[-1]==0:
maxdist=max(maxdist,len(seats)-track1s[-1])
return maxdist
"""