[LeetCode]475. Heaters

最新推荐文章于 2022-01-18 00:15:00 发布

原创最新推荐文章于 2022-01-18 00:15:00 发布 · 405 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

184 篇文章

订阅专栏

本文介绍了解决LeetCode 475. Heaters问题的方法，通过先对房屋和供暖器位置进行排序，再利用双指针技巧遍历房屋列表，寻找每个房屋到供暖器的最短距离，并最终确定所有房屋覆盖所需的最大半径。

[LeetCode]475. Heaters

题目描述

这里写图片描述

思路

从houses角度出发考虑
对于每一个house，找到距离最小的heater，最终的结果就是所有houses中与heater最远的距离
查找优化技巧
现将houses和heaters数组排序，然后对heaters数组使用两个指针，一个指向当前，一个指向下一个
遍历houses，当house对当前heater的距离大于等于对下一个heater的距离的时候，指针移动，直到距离小于，或者next指向heater尾部。
在遍历的同时，维护一个max值，当对一个house得到与heart最小的距离的时候，与当前max值比较，若大于，则更新

代码

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
    int findRadius(vector<int>& houses, vector<int>& heaters) {
        sort(houses.begin(), houses.end());
        sort(heaters.begin(), heaters.end());

        int cur = 0, next = 1, max = 0;
        for (int i = 0; i < houses.size(); ++i) {
            if (heaters.size() == 1){
                if (max < abs(houses[i] - heaters[0])) {
                    max = abs(houses[i] - heaters[0]);
                }
            }
            else {
                while (abs(houses[i] - heaters[cur]) >= abs(houses[i] - heaters[next]) && next < heaters.size() - 1) {
                    ++cur;
                    ++next;
                }
                if (max < min(abs(houses[i] - heaters[cur]), abs(houses[i] - heaters[next])))
                    max = min(abs(houses[i] - heaters[cur]), abs(houses[i] - heaters[next]));
            }

        }
        return max;
    }
};

int main() {
    vector<int> houses = { 1,2,3,2 }, heaters = {1, 2, 3, 2};
    Solution s;
    cout << s.findRadius(houses, heaters) << endl;

    system("pause");
}