在一排座位( 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。
输入测试数据 (每行一个参数)如何理解测试数据?
class Solution {
public:
/**
* @param seats: an array
* @return: the maximum distance
*/
int maxDistToClosest(vector<int> &seats) {
// Write your code here.
int size = seats.size();
int ret = INT_MIN;
vector<int> dp(size);
fill(dp.begin(), dp.end(), INT_MAX);
for(int i = 0; i < size; i++)
{
if(seats[i] == 1)
dp[i] = 0;
else
{
for(int j = i-1; j >= 0; j--)
{
if(seats[j] == 1)
{
dp[i] = dp[i] < abs(i - j) ? dp[i] : abs(i - j);
break;
}
}
for(int j = i+1; j < size; j++)
{
if(seats[j] == 1)
{
dp[i] = dp[i] < abs(i - j) ? dp[i] : abs(i - j);
break;
}
}
ret = ret > dp[i] ? ret : dp[i];
}
cout<<dp[i]<<endl;
}
return ret;
}
};