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 。
提示:
-
1 <= seats.length <= 20000
-
seats
中只含有 0 和 1,至少有一个 0
,且至少有一个 1
。
思路
遍历数组,若元素为1,则距离为0;若元素为0,则距离为坐在上一个座位上的距离+1。
最后对每一个座位,获取左边、右边的距离最小的值,也就是代表着离他最近的人的距离,对这些座位上的最近的距离取最大值。
代码
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
//找两侧连续0的最短长度
int len=seats.size();
vector<int> left(len,len);
vector<int> right(len,len);
for(int i=0;i<len;i++)
{
if(seats[i]==1) left[i]=0;
else if(i>0) left[i]=left[i-1]+1;
}
for(int i=len-1;i>=0;i--)
{
if(seats[i]==1) right[i]=0;
else if(i<len-1) right[i]=right[i+1]+1;
}
int ans=0;
for(int i=0;i<len;i++)
{
if(seats[i]==0)
{
ans=max(ans,min(right[i],left[i]));
}
}
return ans;
}
};