欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Furthest Building You Can Reach

程序员文章站 2022-06-03 15:14:38
...

You are given an integer array heights representing the heights of buildings, some bricks, and some ladders.

You start your journey from building 0 and move to the next building by possibly using bricks or ladders.

While moving from building i to building i+1 (0-indexed),

  • If the current building's height is greater than or equal to the next building's height, you do not need a ladder or bricks.
  • If the current building's height is less than the next building's height, you can either use one ladder or (h[i+1] - h[i]) bricks.

Return the furthest building index (0-indexed) you can reach if you use the given ladders and bricks optimally.

 

Example 1:

Furthest Building You Can Reach

Input: heights = [4,2,7,6,9,14,12], bricks = 5, ladders = 1
Output: 4
Explanation: Starting at building 0, you can follow these steps:
- Go to building 1 without using ladders nor bricks since 4 >= 2.
- Go to building 2 using 5 bricks. You must use either bricks or ladders because 2 < 7.
- Go to building 3 without using ladders nor bricks since 7 >= 6.
- Go to building 4 using your only ladder. You must use either bricks or ladders because 6 < 9.
It is impossible to go beyond building 4 because you do not have any more bricks or ladders.

思路:greedy思路,就是用maxheap记录当前用掉的bricks,因为ladder可以填充任意的高度,所以很珍贵,首先用bricks填充,然后如果不够了,那么就用ladder去填充之前用过的最大的bricks。

class Solution {
    public int furthestBuilding(int[] heights, int bricks, int ladders) {
        PriorityQueue<Integer> maxheap = new PriorityQueue<Integer>((a, b) -> (b - a));
        for(int i = 0; i < heights.length; i++) {
            if(i + 1 < heights.length && heights[i] >= heights[i + 1]) {
                continue;
            } 
            if(i + 1 < heights.length && heights[i] < heights[i + 1]) {
                int diff = heights[i + 1] - heights[i];
                // if bricks is enough, first use bricks.
                if(bricks >= diff) {
                    bricks -= diff;
                    maxheap.offer(diff);
                } else {
                    // not enough bricks, try to use ladder fill largest bricks gap;
                    if(ladders > 0) {
                        if(!maxheap.isEmpty()) {
                            bricks += maxheap.poll();
                            // if bricks is enough, first use bricks. 
                            if(bricks >= diff) {
                               bricks -= diff;
                               maxheap.offer(diff); 
                            }
                        }
                        ladders--;
                    } else {
                        return i;
                    }
                }
            }
        }
        return heights.length - 1;
    }
}