题目:
给你一个数组 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.坐在第一个位置,2.坐在最后一个位置,3.坐在两个人中间,分别把这三种情况下的空位置数,求出,最后取最大的即可,具体看代码
代码:
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
//用来表示前一个坐人的位置下标
int pre=0 ;
int len = seats.size();
//记录第一个人前面的空位置数
int count1 = 0;
//记录中间两个人之间的空位置数
int count2 = 0;
//找到第一个坐人的位置,付给pre
for (int i = 0; i < len; i++) {
if (seats[i] == 1) {
pre = i;
break;
}
}
//空位个数
if (pre != 0) count1 = pre;
//只有最后一个位置有人
if (pre == len - 1) return len - 1;
//在上次的基础上往后遍历
for (int i = pre + 1; i < len; i++) {
if (seats[i] == 1) {
if (i - pre-1 > count2) {
count2 = i - pre - 1;
}
pre = i;
}
}
//if (pre < len - 1) count3 = len - 1 - pre;
count1 = (count1 > (len - 1 - pre) ? count1 : (len - 1 - pre));
//两人中间空位置数为偶数或奇数,有一定区别
if (count2 % 2 == 0) {
return max(count2 / 2, count1);
}
return max(count2 / 2 + 1, count1);
}
};