题目:
在一排座位( 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的个数,
1、如果在两边的话,有几个零,最大距离就是几,
2、在中间的情况有两种,偶数个零,则距离就是个数除以2,奇数个零,则距离就是个数除以2+1
代码:
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
long len = seats.size();
if(len == 1)
return 0;
int i = 0, cnt = 0, ret = 0;
for(; i < len && seats[i] == 0; i++) cnt++;
ret = max(cnt, ret);
cnt = 0;
int j = len - 1;
for(; j >= 0 && seats[j] == 0; j--) cnt++;
ret = max(cnt, ret);
while(i < j){
for(;i < j && seats[i] == 1; i++);
cnt = 0;
for(; i < j && seats[i] == 0; i++)cnt++;
if(cnt % 2 == 0)
ret = max(cnt/2, ret);
else
ret = max(cnt/2 + 1, ret);
i++;
}
return ret;
}
};
复杂度 o(n);