LeetCode 475. 供暖器（C++、python）

冬季已经来临。你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。

现在，给出位于一条水平线上的房屋和供暖器的位置，找到可以覆盖所有房屋的最小加热半径。

所以，你的输入将会是房屋和供暖器的位置。你将输出供暖器的最小加热半径。

说明:

给出的房屋和供暖器的数目是非负数且不会超过 25000。
给出的房屋和供暖器的位置均是非负数且不会超过10^9。
只要房屋位于供暖器的半径内(包括在边缘上)，它就可以得到供暖。
所有供暖器都遵循你的半径标准，加热的半径也一样。

示例 1:

输入: [1,2,3],[2]
输出: 1
解释: 仅在位置2上有一个供暖器。如果我们将加热半径设为1，那么所有房屋就都能得到供暖。

示例 2:

输入: [1,2,3,4],[1,4]
输出: 1
解释: 在位置1, 4上有两个供暖器。我们需要将加热半径设为1，这样所有房屋就都能得到供暖。

C++

class Solution {
public:
    int findRadius(vector<int>& houses, vector<int>& heaters) 
    {
        int m=houses.size();
        int n=heaters.size();
        sort(houses.begin(),houses.end());
        sort(heaters.begin(),heaters.end());
        vector<int> right(m,INT_MAX);
        int i=0;
        int j=0;
        while(i<m && j<n)
        {
            if(houses[i]<=heaters[j])
            {
                right[i]=heaters[j]-houses[i];
                i++;
            }
            else
            {
                j++;
            }
        }
        vector<int> left(m,INT_MAX);
        i=m-1;
        j=n-1;
        while(i>=0 && j>=0)
        {
            if(houses[i]>=heaters[j])
            {
                left[i]=houses[i]-heaters[j];
                i--;
            }
            else
            {
                j--;
            }
        }
        for(int i=0;i<m;i++)
        {
            left[i]=min(left[i],right[i]);
        }
        return *max_element(left.begin(),left.end());        
    }
};

python

class Solution:
    def findRadius(self, houses: List[int], heaters: List[int]) -> int:
        val=max(max(houses),max(heaters))
        m=len(houses)
        n=len(heaters)
        houses=sorted(houses)
        heaters=sorted(heaters)
        i=0
        j=0
        right=[val for k in range(m)]
        while i<m and j<n:
            if houses[i]<=heaters[j]:
                right[i]=heaters[j]-houses[i]
                i+=1
            else:
                j+=1
        i=m-1
        j=n-1
        left=[val for k in range(m)]
        while i>=0 and j>=0:
            if houses[i]>=heaters[j]:
                left[i]=houses[i]-heaters[j]
                i-=1
            else:
                j-=1
        for i in range(m):
            left[i]=min(left[i],right[i])
        return max(left)

LeetCode 475. 供暖器（C++、python）

猜你喜欢