LeetCode--最后一块石头的重量(堆 贪心)

                                   最后一块石头的重量

有一堆石头,每块石头的重量都是正整数。

每一回合,从中选出两块最重的石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:

如果 x == y,那么两块石头都会被完全粉碎;

如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。

最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 0。 

提示:

1 <= stones.length <= 30

1 <= stones[i] <= 1000

public class lastStoneWeight {
    public static void main(String[] args) {
        int[] stones = {2,7,4,1,8,1};
        lastStoneWeight(stones);
    }
    public static int lastStoneWeight(int[] stones) {
        if(stones == null){
            return 0;
        }
        if(stones.length< 2){
            return stones[0];
        }
        int length=stones.length;
        while(length>=2){
            Arrays.sort(stones);
            if(stones[length-2]==stones[length-1]) {
                length -= 2;
            }
            else{
                stones[length-2]=stones[length-1]-stones[length-2];
                length--;

            }
        }
        if(length==1) {
            return stones[0];
        }
        else {
            return 0;
        }
    }
}

堆排序

  • 初始化最大堆
  • 循环(直至Math.max(stones[1], stones[2])==0
    • 按要求修改堆顶两个元素

重新调整最大堆

public int lastStoneWeight(int[] stones) {
    int len = stones.length;
    if (len==1) {
        return stones[0];
    }else if (len==2) {
        return Math.abs(stones[0]-stones[1]);
    }else {
        for (int i = len/2-1; i >= 0 ; i--) {
            maxHeap(stones, i, len-1);
        }
        int temp;
        while ((temp=Math.max(stones[1], stones[2]))!=0) {
            stones[0]-=temp;
            if (stones[1]>stones[2]) {
                stones[1]=0;
            }else {
                stones[2]=0;
            }
            for (int i = 2; i >= 0 ; i--) {
                maxHeap(stones, i, len-1);
            }
        }
        return stones[0];
    }
}
public void maxHeap(int[] nums, int root, int end) {
    int l=2*root+1;
    int temp=nums[root];
    while (l<=end && (nums[l]>temp || (l+1<=end && nums[l+1]>temp))) {
        if (l+1<=end && nums[l+1]>nums[l]) {
            l++;
        }
        nums[root]=nums[l];
        nums[l]=temp;
        root=l; l=2*root+1;
        temp=nums[root];
    }
}

使用现成的大顶堆

/**
     * java大顶堆
     * 执行用时 :43 ms, 在所有 Java 提交中击败了10.83%的用户
     * 内存消耗 :34.4 MB, 在所有 Java 提交中击败了100.00%的用户
     */
    public static int lastStoneWeight3(int[] stones) {
        PriorityQueue<Integer> pq = new PriorityQueue<>((i1, i2) -> i2 - i1);
        for (int x: stones) {
            pq.offer(x);
        }
        while (pq.size() > 1) {
            int remaining = pq.poll() - pq.poll();
            if (remaining != 0) {
                pq.offer(remaining);
            }
        }
        return pq.isEmpty() ? 0 : pq.poll();
    }

双栈实现最大堆

/**
     * 双栈实现最大堆
     * 执行用时 : 7 ms, 在所有 Java 提交中击败了15.14%的用户
     * 内存消耗 :34 MB, 在所有 Java 提交中击败了100.00%的用户
     */
    public static int lastStoneWeight(int[] stones) {
        Stack<Integer> max = new Stack();
        Stack<Integer> temp = new Stack<>();
        for (int x: stones) {
            while (!max.isEmpty() && max.peek() > x) {
                temp.push(max.pop());
            }
            max.push(x);
            while (!temp.isEmpty()) {
                max.push(temp.pop());
            }
        }
        while (max.size() > 1) {
            int remaining = max.pop() - max.pop();
            if (remaining != 0) {
                while (!max.isEmpty() && max.peek() > remaining) {
                    temp.push(max.pop());
                }
                max.push(remaining);
                while (!temp.isEmpty()) {
                    max.push(temp.pop());
                }
            }
        }
        return max.isEmpty() ? 0 : max.pop();
    }
}

 

 

 

全部评论

相关推荐

已经入职字节快一个月了,突然想起来之前那段时间的面经没有发,先发一下timeline吧。Tiktok&nbsp;内容安全平台(人才库电话捞我):电话10.28&nbsp;-&gt;&nbsp;一面10.30(我觉得你跟我们组业务挺match的,然后过了三天问hr挂了,应该是别人流程更快)阿里淘天:投递11.11-&gt;约面11.12-&gt;一面11.14(问得很简单,30分钟,手撕八股全过无后续)Kpi面腾讯wxg&nbsp;微信小程序:投递11.13&nbsp;-&gt;约面11.14-&gt;&nbsp;一面11.17&nbsp;(究极无敌拷打,问我多模态大模型涉及的算法?但是人很好)-&gt;11.19流程终止小红书&nbsp;风控平台:投递11.16&nbsp;—约面11.17&nbsp;&nbsp;-&gt;一面11.18&nbsp;(抽象的面试官,面试感觉一般,问得前端网页安全相关的,确实没准备)-&gt;11.20挂百度&nbsp;百家号:投递11.14&nbsp;—&gt;约面11.14&nbsp;-&gt;一面11.14(当场约2面)-&gt;二面11.24-&gt;口头告知offer,&nbsp;拒绝(原因是业务不太好)美团&nbsp;技术平台投递11.17&nbsp;-&gt;&nbsp;约面(忘记了,没多久)&nbsp;-&gt;一面11.19&nbsp;-&gt;二面11.21&nbsp;(字节offer不想面了)快手&nbsp;电商业务投递11.17&nbsp;-&gt;&nbsp;约面11.18&nbsp;-&gt;一面11.19&nbsp;-&gt;&nbsp;二面11.21(拒了)腾讯wxg&nbsp;微信支付(被捞):(直接发面试邮件)技术一面12.05&nbsp;-&gt;技术二面12.11&nbsp;-&gt;技术三面12.17&nbsp;-&gt;&nbsp;hr面已拒绝(了解业务后拒绝,但是有转正hc,感觉还蛮可惜)字节跳动&nbsp;xxxx:东家就不放具体的时间线了,大概是面完第二天就能知道结果,除了有几天ld请假了没填面评。不去wxg还有个原因是还在期末周,深圳学校来回太麻烦了,至少现在在的组感觉能学到很多的东西,自己的选择应该也没错。还是感概一下,一年前大二的时候投简历海投基本上石沉大海,无论大小厂约面比例很少。现在基本上投了就有面试,还都是以前梦寐以求的大厂,现在自己也有了更多的选择,也没有投太多简历。也感谢上一段实习的经历,很有意思的项目,无论是字节,腾讯,还是美团基本都有聊这个项目。面经需要等一下,也许等周末有空了再发给各位uu,感兴趣可以关注一下~有想要交流学习的同学也可以私信我,目前人在北京大钟寺~,可以找搭子~
正能量的牛可乐:这么多大厂面试下来,不仅摸清了不同公司的面试风格,还能精准避雷业务不匹配的岗位,血赚
实习简历求拷打
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务