Leetcode 849. 到最近的人的最大距离
家人们谁懂啊,七夕节解决小便池如何择位
题目描述
难度:中等
题目内容:
给你一个数组 seats 表示一排座位,其中 seats[i] = 1 代表有人坐在第 i 个座位上,seats[i] = 0 代表座位 i 上是空的(下标从 0 开始)。
至少有一个空座位,且至少有一人已经坐在座位上。
亚历克斯希望坐在一个能够使他与离他最近的人之间的距离达到最大化的座位上。
返回他到离他最近的人的最大距离。
男童鞋们懂得都懂,这不就是尽可能地离别人远一点嘘嘘吗
示例 1:
输入:seats = [1,0,0,0,1,0,1]
输出:2
解释:
如果亚历克斯坐在第二个空位(seats[[2]])上,他到离他最近的人的距离为 2 。
如果亚历克斯坐在其它任何一个空位上,他到离他最近的人的距离为 1 。
因此,他到离他最近的人的最大距离是 2 。
示例 2:
输入:seats = [1,0,0,0]
输出:3
解释:
如果亚历克斯坐在最后一个座位上,他离最近的人有 3 个座位远。
这是可能的最大距离,所以答案是 3 。
示例 3:
输入:seats = [0,1]
输出:1
提示:
2 <= seats.length <= 2 * 104
seats[i] 为 0 或 1
至少有一个 空座位
至少有一个 座位上有人
解答思路
1.首先考虑普通情况,即两端都有人时
首先我们遍历一遍seats数组,并使用seat数组记录有人的位置。
示例:
seats = [1,0,1,0,1,0,0,0,1,0,1]
seat = [0,2,4,8,10]
其次遍历seat数组并计算相邻有人位置的中间距离,并用dist数组存储
示例:
seat = [0,2,4,8,10]
dist = [1,1,2,1]
选择中间距离中的最大距离即为所求距离
示例:
dist = [1,1,2,1]
max(dist) = 2
2.当两端中某一端或两端无人时
示例:
seats = [0,0,0,0,0,0,0,0,1,0,0]
此时我们需要人为地端点处成为可能的目标位置,这里引用两个虚拟位置变量first和last(因为seats[first] 和 seats[last] 无法访问),并将first和last值分别插入到seat数组两端,其中
seat = [8]
first = seat[0] * (-1) = -8
last = (len(seats) - 1 + len(seats) - 1
= 2*(len(seats) - 1)- seat[-1])
= 12
插入后
seat = [-8,8,12]
dist = [8,2]
即 max(dist) = 8
代码验证
class Solution(object):
def maxDistToClosest(self, seats):
"""
:type seats: List[int]
:rtype: int
"""
seat = []
for i in range(0, len(seats)):
if seats[i] == 1:
seat.append(i)
first = [seat[0] * (-1)]
seat = first+seat
last = [(len(seats) - 1 - seat[-1])+len(seats)-1]
seat = seat+last
dist = []
for i in range(1, len(seat)):
dist.append(int((seat[i] - seat[i - 1]) / 2))
return max(dist)
结果:
总结:
感觉有点空间换时间的意思,希望能优化一下吧