题目描述
在一排座位( 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
。
解题思路
和leetcode 868 二进制间距相似https://mp.csdn.net/postedit/89252325,找出被1个零或多个零分割的1,然后考虑让亚历克斯坐在中间位置;但只按上述方法找位置会遗漏000011101000000的情况,所以要记录第一个1和最后一个1出现的位置
int maxDistToClosest(vector<int>& seats) {
int pre = 0,post = 1,len = seats.size();
int ans = 0,first = INT_MAX,last = 0;
while(post < len){
if(seats[pre] == 0) pre = post;
else{
if(seats[post] == 1){
if(post != pre+1){
int ya = (post+pre)/2;
int tmp = min(ya-pre,post-ya);
ans = max(ans,tmp);
}
pre = post;
}
}
if(seats[pre] == 1) first = min(first,pre);
if(seats[post] == 1) last = max(last,post);
post++;
}
if(first == len-1) return first;
ans = max(ans,first-0);
ans = max(ans,len-last-1);
return ans;
}