题目来源
题目描述
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
}
};
题目解析
其实就是计算最大连续0的长度,头和尾特殊计算.
三种情况:
- 左边全都可以坐,例如:[0,0,0,0,0,0,0,1]那就坐最左边的位置;
- 右边全都可以坐,例如[1,0,0,0,0,0,0,0],那就坐最右边的位置;
- 左右两边都有人,那就看哪一段空的座位最多,例如[1,0,0,0,0,0,1,0,1],有两段,坐第一段最中间。
最大距离有三种可能性:
- 第一个1到最左边的距离
- 相邻的1中间位置到彼此的距离
- 最后一个1到最右边的距离
计算座位到最近的人的最大距离
思路:
- 令left[i]为座位i到坐在i左边的人的最近距离。同理right[i]为座位i到做到i右边的人的最近距离。那么该座位到最近的人的距离为:min(left[i],right[i])
算法:
- 如果i左边的位置是空的,那么left[i] == left[i - 1] + 1,否则right[i] = 0
- right[i] 的计算方法类似。
实现:
class Solution {
public:
int maxDistToClosest(vector<int>& seats) {
int N = seats.size();
std::vector<int> left(N, N), right(N, N);
for (int i = 0; i < N; ++i) {
if(seats[i] == 1){
left[i] == 0;
}else if(i > 0){
left[i] = left[i - 1] + 1;
}
}
for (int i = N-1; i >= 0; --i) {
if (seats[i] == 1) right[i] = 0;
else if (i < N-1) right[i] = right[i+1] + 1;
}
int ans = 0;
for (int i = 0; i < N; ++i) {
if(seats[i] == 0){
ans = std::max(ans, std::min(left[i], right[i]));
}
}
return ans;
}
};