在一排座位( 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。
分析:
最远距离和 0 个数之间的距离:
0的个数 | 最远距离 |
---|---|
1 | 1 |
2 | 1 |
3 | 2 |
4 | 2 |
.. | .. |
n | (n+1)/2 |
最远距离 = (0 的个数+1)/ 2
而 0 的个数 = 当前 1 的位置 - 上一个 1 的位置 - 1
所以 最远距离 = (当前 1 的位置(i)- 上一个 1 的位置(flag))/ 2
考虑 0 开头和 0 结尾的特殊情况:
0 的个数就是最远距离(只有一边有人)
class Solution {
public int maxDistToClosest(int[] seats) {
int max = 0;
int flag = -1;
for (int i = 0; i < seats.length; i++) {
if (seats[i] == 1) {
if (flag == -1)
max = i;
else
max = Integer.max((i - flag) / 2, max);
flag = i;
}
}
return Integer.max(max, seats.length - flag - 1);
}
}