LeetCode849:到最近的人的最大距离

在一排座位( 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. 1 <= seats.length <= 20000
  2. 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;

}

猜你喜欢

转载自blog.csdn.net/qq_36214481/article/details/86134705
今日推荐