Every day a Leetcode
题目来源:1642. 可以到达的最远建筑
解法1:反悔贪心
在移动的过程中,我们会需要若干次需要使用砖块或者梯子的情况。假设当前我们需要移动到下一建筑物,但必须使用 1 架梯子或者 Δh 个砖块,那么我们如何抉择是使用梯子还是砖块呢?
我们可以用贪心的思路来想这个问题。「梯子」相当于一次性的无限量砖块,那么我们一定是把梯子用在刀刃上。也就是说,如果我们有 ladders 架梯子,那么我们会在 Δh 最大的那 ladders 次使用梯子,而在剩余的情况下使用砖块。
这样一来,我们就可以得到正确的算法了:我们使用优先队列(小根堆)实时维护不超过 ladders 个最大的 Δh,这些就是使用梯子的地方。对于剩余的 Δh,我们需要使用砖块,因此需要对它们进行累加,如果某一时刻这个累加值超过了砖块的数目 bricks,那么我们就再也无法移动了。
代码:
/*
* @lc app=leetcode.cn id=1642 lang=cpp
*
* [1642] 可以到达的最远建筑
*/
// @lc code=start
class Solution
{
public:
int furthestBuilding(vector<int> &heights, int bricks, int ladders)
{
int n = heights.size();
priority_queue<int, vector<int>, greater<int>> pq;
int sumH = 0;
for (int i = 0; i < n - 1; i++)
{
int deltaH = heights[i + 1] - heights[i];
if (deltaH > 0)
{
pq.push(deltaH);
if (pq.size() > ladders)
{
sumH += pq.top();
pq.pop();
}
if (sumH > bricks)
return i;
}
}
return n - 1;
}
};
// @lc code=end
结果:
复杂度分析:
时间复杂度:O(nlogl),其中 n 是建筑物的数量,l 是梯子的数量。
空间复杂度:O(l),其中 n l 是梯子的数量。