leetcode 849. 到最近的人的最大距离(Maximize Distance to Closest Person)
题目描述:
在一排座位( 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
。
解法:
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
//0 ==> 1
//00 ==> 1
//000 ==> 2
int sz = seats.size();
int i = 0, j = 0;
int res = 0;
while(i < sz){
while(j < sz && seats[j] == 0){
j++;
}
if(i == j){
// is one
j++;
}else{
if(i == 0 || j == sz){
res = max(res, j - i);
}else{
res = max(res, (j - i+1)/2);
}
}
i = j;
}
return res;
}
};