1. 程式人生 > >Leetcode849.Maximize Distance to Closest Person到最近的人的最大距離

Leetcode849.Maximize Distance to Closest Person到最近的人的最大距離

在一排座位( 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。
class Solution {
public:
    int maxDistToClosest(vector<int>& seats) {
        int len = seats.size();
        if(len <= 1)
            return 0;
        int MAX = 0;
        if(seats[0] == 0)
        {
            int cnt = 0;
            for(int i = 0; i < len; i++)
            {
                if(seats[i] == 0)
                    cnt++;
                else
                    break;
            }
            MAX = max(MAX, cnt);
        }
        if(seats[len - 1] == 0)
        {
            int cnt = 0;
            for(int i = len - 1; i >= 0; i--)
            {
                if(seats[i] == 0)
                    cnt++;
                else
                    break;
            }
            MAX = max(MAX, cnt);
        }
        int cnt = 0;
        for(int i = 1; i < len - 1; i++)
        {
            if(seats[i] == 0)
            {
                cnt++;
                MAX = max((cnt + 1) / 2, MAX);
            }
            else
            {
                cnt = 0;
            }
        }
        return MAX;
    }
};