题目描述
在一排座位( seats)中,1 代表有人坐在座位上,0 代表座位上是空的。至少有一个空座位,且至少有一人坐在座位上。
小明希望坐在一个能够使他与离他最近的人之间的距离达到最大化的座位上,返回他到离他最近的人的最大距离。
注意:
1.1 <= seats.length <= 20000
2. seats 中只含有 0 和 1,至少有一个 0,且至少有一个 1。
示例
输入:[1,0,0,0,1,0,1]
输出:2
解释:如果小明坐在第二个空位(seats[2])上,他到离他最近的人的距离为 2 。如果小明坐在其它任何一个空位上,他到离他最近的人的距离为 1 。因此,他到离他最近的人的最大距离是 2 。
输入:[1,0,0,0]
输出:3
解释: 如果小明坐在最后一个座位上,他离最近的人有 3 个座位远。这是可能的最大距离,所以答案是 3 。
解答
首先找到全部的1的索引;接着统计连续0的数量,并找到最长的连续0数目
坐座位有三种情况:
- 坐在最左边,距离为到第一个1的距离
- 坐在最右边,距离为到最后一个1的距离
- 坐在中间(某两个1之间),坐在中间必然是坐到最长连续0的中间,这又有两种情况:如果是奇数,则坐最中间,距离为n/2(取整数部分);如果是偶数,则会偏(如,001000),距离为n/2-1
比较这三种情况,取最大值
class Solution(object):
def maxDistToClosest(self, seats):
"""
:type seats: List[int]
:rtype: int
"""
length = len(seats)
# 找到1的索引
index = list()
for i in range(0,length):
if seats[i]==1:
index.append(i)
max_empty = 0
len_ind = len(index)
for i in range(1,len_ind):
empty = index[i]-index[i-1]-1
if empty>max_empty:
max_empty = empty
# 三种情况
# 坐在最左边
# 坐在某两个1中间
# 坐在最右边
# 比较这三种情况下哪个距离最大
if max_empty%2==0: # 中间空了偶数个座位
return max([index[0],max_empty/2,length-1-index[-1]])
else:
return max([index[0],max_empty/2+1,length-1-index[-1]])
# 数组内元素是int,做除法只取整数部分