475. 供暖器

看了题解之后在纸上画了画才写出来的。可以假设对于每个house左右都有加热器,那么这个house的半径就是min(左加热器,右加热器)。ans则是所有house半径的最大值。

初始化ans为0。将heaters从小到大排序,遍历houses,对于houses中的每个house,找到第1个大于house的位置:

1. 如果这个位置index等于0,说明heater的第1个数就大于house 475. 供暖器这个house就只有这1个加热器(右加热器),这个半径就等于475. 供暖器,计算ans = max(ans, 这个半径)

2. 如果这个位置index等于heaters.size(),说明找到最后也没有在heaters中找到大于house的heater 475. 供暖器这个house就只有最后一个加热器离它最近(左加热器),这个半径就等于475. 供暖器,取ans和这个半径的最大值

3. 然后就是一般情况,house的右边加热器为475. 供暖器,house左边的加热器为475. 供暖器,这个house的最小半径就是475. 供暖器,这个最小半径和ans取最大值。

4. 最后返回ans即可

class Solution {
public:
    int findRadius(vector<int>& houses, vector<int>& heaters) {
        sort(heaters.begin(),heaters.end());
        int ans = 0;
        for (int house:houses){
            // 在heaters中查找第1个大于house的位置
            int uppos = upper_bound(heaters.begin(),heaters.end(),house) - heaters.begin();
            if (uppos ==0 ){
                // 只有一个加热器
                ans = max(ans,heaters[uppos]-house);
            }
             else if (uppos == heaters.size()){
                ans = max(ans,house - heaters[uppos-1]);
            }
            else{
                // 有2个加热器
                int lowpos = uppos-1;
                int tmpans = min(house-heaters[lowpos],heaters[uppos]-house);
                ans = max(ans,tmpans);
            }
        }
        return ans;
    }
};

475. 供暖器

 

上一篇:Elasticsearch启动常见问题


下一篇:codeforces 3D Least Cost Bracket Sequence