在一排座位( 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
。
解析:
该题目需要注意数组开头连续0和数组结尾连续0的特殊性,它们不需要做除2的操作,因此需要单独处理。数组中部的0需要做+1除2的操作,最后对比三处的大小,即可得出最终结果。
代码:
int maxDistToClosest(vector<int>& seats)
{
int size = seats.size();
int start = 0;
int end = size - 1;
while (seats[start] == 0)start++;
while (seats[end] == 0)end--;
int startNumOf0 = start;//以0开头的连续0个数
int endNumOf0 = size - end - 1;//以0结尾的连续0个数
int midNumOf0 = 0;
int maxNumOf0 = 0;
int indexOfMid = 0;
for (int i = start; i <= end; i++)
{
if (seats[i] == 1)
{
if (midNumOf0 > maxNumOf0)
{
maxNumOf0 = midNumOf0;
}
midNumOf0 = 0;
}
else
{
midNumOf0++;
}
}
//比较三处大小
if (startNumOf0 >= endNumOf0&&startNumOf0 >= (maxNumOf0+1)/2)
return startNumOf0;
else if (endNumOf0 > startNumOf0&&endNumOf0 >= (maxNumOf0 + 1) / 2)
return endNumOf0;
else
return (maxNumOf0 + 1) / 2;
}