题目
难度:★★☆☆☆
类型:数组
在一排座位( seats)中,1 代表有人坐在座位上,0 代表座位上是空的。
至少有一个空座位,且至少有一人坐在座位上。
亚历克斯希望坐在一个能够使他与离他最近的人之间的距离达到最大化的座位上。
返回他到离他最近的人的最大距离。
提示
1 <= seats.length <= 20000
seats 中只含有 0 和 1,至少有一个 0,且至少有一个 1。
示例
示例 1
输入:[1,0,0,0,1,0,1]
输出:2
解释:
如果亚历克斯坐在第二个空位(seats[2])上,他到离他最近的人的距离为 2 。
如果亚历克斯坐在其它任何一个空位上,他到离他最近的人的距离为 1 。
因此,他到离他最近的人的最大距离是 2 。
示例 2
输入:[1,0,0,0]
输出:3
解释:
如果亚历克斯坐在最后一个座位上,他离最近的人有 3 个座位远。
这是可能的最大距离,所以答案是 3 。
解答
首先,我们需要获得所有人所在的位置persons。
我们需要考虑三种情况:
左边全都可以坐,例如:[0,0,0,0,0,0,0,1],可以考虑坐最左边的位置,这时与最近的人的距离是第一个人所在的位置person[0];
右边全都可以坐,例如[1,0,0,0,0,0,0,0],可以考虑坐在最右边的位置,这时与最近的人的距离取决于座位总长度和最后一个人所在的位置:len(seats)-person[-1]+1;
左右两边都有人,那就看哪一段空的座位最多,我们将两个相邻的人中间的所有座位成为一段,例如[1,0,0,0,0,0,1,0,1],有两段,第一段最长,因此坐第一段最中间,这时与最近的人的距离为该段长度的一半,我们的计算需要考虑所有的段,然后取其最大值。
以上三种情况需要求最大值,即为最终结果。
class Solution:
def maxDistToClosest(self, seats):
"""
:param seats: List[int]
:return: int
"""
persons = [p for p, seat in enumerate(seats) if seat == 1]
return max(persons[0], len(seats) - 1 - persons[-1], max([(persons[r] - persons[r - 1]) // 2 for r in range(1, len(persons))]+[0]))
如有疑问或建议,欢迎评论区留言~