comments | difficulty | edit_url | tags | |
---|---|---|---|---|
true |
中等 |
|
给你一个数组 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
- 至少有一个 空座位
- 至少有一个 座位上有人
我们定义两个变量
然后遍历数组
最后返回
时间复杂度
class Solution:
def maxDistToClosest(self, seats: List[int]) -> int:
first = last = None
d = 0
for i, c in enumerate(seats):
if c:
if last is not None:
d = max(d, i - last)
if first is None:
first = i
last = i
return max(first, len(seats) - last - 1, d // 2)
class Solution {
public int maxDistToClosest(int[] seats) {
int first = -1, last = -1;
int d = 0, n = seats.length;
for (int i = 0; i < n; ++i) {
if (seats[i] == 1) {
if (last != -1) {
d = Math.max(d, i - last);
}
if (first == -1) {
first = i;
}
last = i;
}
}
return Math.max(d / 2, Math.max(first, n - last - 1));
}
}
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
int first = -1, last = -1;
int d = 0, n = seats.size();
for (int i = 0; i < n; ++i) {
if (seats[i] == 1) {
if (last != -1) {
d = max(d, i - last);
}
if (first == -1) {
first = i;
}
last = i;
}
}
return max({d / 2, max(first, n - last - 1)});
}
};
func maxDistToClosest(seats []int) int {
first, last := -1, -1
d := 0
for i, c := range seats {
if c == 1 {
if last != -1 {
d = max(d, i-last)
}
if first == -1 {
first = i
}
last = i
}
}
return max(d/2, max(first, len(seats)-last-1))
}
function maxDistToClosest(seats: number[]): number {
let first = -1,
last = -1;
let d = 0,
n = seats.length;
for (let i = 0; i < n; ++i) {
if (seats[i] === 1) {
if (last !== -1) {
d = Math.max(d, i - last);
}
if (first === -1) {
first = i;
}
last = i;
}
}
return Math.max(first, n - last - 1, d >> 1);
}