Leetcode 849. 到最近的人的最大距离
题目
在一排座位( 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 。
提示:
- 2 <= seats.length <= 20000
- seats 中只含有 0 和 1,至少有一个 0,且至少有一个 1。
题解
从左到右、从右到左记录最新有人的座位下标。遍历每个空位,找距离左边有人座位+右边有人座位距离最大的位置。
详细过程见代码
代码
int maxDistToClosest(vector<int>& seats) {
int n = seats.size();
vector<long> L(n,0); //L[i],记录i左边最近的有人座位的下标
vector<long> R(n,0); //R[i],记录i左边最近的有人座位的下标
int place,i=0;
while(i < n){
if(seats[i] == 1){
place = i;
L[i] = i;
break;
}
L[i] = INT_MIN;
i++;
}
for(i++; i<n; i++){
if(seats[i] == 1){
place = i;
}
L[i] = place;
}
i = n-1;
while(i >= 0){
if(seats[i] == 1){
place = i;
R[i] = i;
break;
}
R[i] = INT_MAX;
i--;
}
for(i--; i>=0; i--){
if(seats[i] == 1){
place = i;
}
R[i] = place;
}
long ans=0;
for(int i=0; i<n; i++){
ans = max(ans,min(R[i]-i,i-L[i]));
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximize-distance-to-closest-person
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。