题目
在一排座位( seats)中,1 代表有人坐在座位上,0 代表座位上是空的。
至少有一个空座位,且至少有一人坐在座位上。
亚历克斯希望坐在一个能够使他与离他最近的人之间的距离达到最大化的座位上。
返回他到离他最近的人的最大距离。
示例 1:
输入:[1,0,0,0,1,0,1]
输出:2
解释:
如果亚历克斯坐在第二个空位(seats[2])上,他到离他最近的人的距离为 2 。
如果亚历克斯坐在其它任何一个空位上,他到离他最近的人的距离为 1 。
因此,他到离他最近的人的最大距离是 2 。
示例 2:
输入:[1,0,0,0]
输出:3
解释:
如果亚历克斯坐在最后一个座位上,他离最近的人有 3 个座位远。
这是可能的最大距离,所以答案是 3 。
提示:
1 <= seats.length <= 20000
seats 中只含有 0 和 1,至少有一个 0,且至少有一个 1。
思路
- 将seats中为1的索引找出来
- 求取所有两个1的索引间距的一半,选取最大值
- 判断一开始不是1,和最后不是1的情况
class Solution:
def maxDistToClosest(self, seats):
"""
:type seats: List[int]
:rtype: int
"""
lens = len(seats)
#将seats中为1的索引找出来
oneIndex = [i for i in range(lens) if seats[i] == 1]
lenOne = len(oneIndex)
#求取所有两个1的索引间距的一半,选取最大值
maxR = max([oneIndex[i+1]-oneIndex[i] for i in range(lenOne-1)]+[0])//2
if seats[0] == 0:
maxR = max(maxR,oneIndex[0]-0)
if seats[-1] == 0:
maxR = max(maxR,lens-1-oneIndex[-1])
return maxR