给你一个数组 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
- 至少有一个 空座位
- 至少有一个 座位上有人
package Solution849;
import java.util.Arrays;
class Solution {
public int maxDistToClosest(int[] seats) {
int max = 0;
int count = 0;
for (int i = 0; i < seats.length; i++) {
if (seats[i] == 0) {
count++;
if (i == seats.length - 1) {
if (count > max) {
max = count;
return max;
}
}
} else {
if (count > max) {
max = count;
}
count = 0;
}
}
int countleft = 0;
for (int i = 0; i < seats.length; i++) {
if (seats[i] == 0) {
countleft++;
} else {
break;
}
}
int countright = 0;
for (int i = seats.length - 1; i > -1; i--) {
if (seats[i] == 0) {
countright++;
} else {
break;
}
}
int maxedge = Math.max(countleft, countright);
int maxmid = 0;
if (max % 2 == 1) {
maxmid = max / 2 + 1;
} else {
maxmid = max / 2;
}
return maxedge > maxmid ? maxedge : maxmid;
}
public static void main(String[] args) {
Solution sol = new Solution();
int[] seats = { 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0 };
System.out.println(sol.maxDistToClosest(seats));
}
}