算法41:掉落的方块(力扣699题)----线段树

news2024/11/16 19:30:56

题目:https://leetcode.cn/problems/falling-squares/description/

在二维平面上的 x 轴上,放置着一些方块。

给你一个二维整数数组 positions ,其中 positions[i] = [lefti, sideLengthi] 表示:第 i 个方块边长为 sideLengthi ,其左侧边与 x 轴上坐标点 lefti 对齐。

每个方块都从一个比目前所有的落地方块更高的高度掉落而下。方块沿 y 轴负方向下落,直到着陆到 另一个正方形的顶边 或者是 x 轴上 。一个方块仅仅是擦过另一个方块的左侧边或右侧边不算着陆。一旦着陆,它就会固定在原地,无法移动。

在每个方块掉落后,你必须记录目前所有已经落稳的 方块堆叠的最高高度 。

返回一个整数数组 ans ,其中 ans[i] 表示在第 i 块方块掉落后堆叠的最高高度。

示例 1:

输入:positions = [[1,2],[2,3],[6,1]]
输出:[2,5,5]
解释:
第 1 个方块掉落后,最高的堆叠由方块 1 组成,堆叠的最高高度为 2 。
第 2 个方块掉落后,最高的堆叠由方块 1 和 2 组成,堆叠的最高高度为 5 。
第 3 个方块掉落后,最高的堆叠仍然由方块 1 和 2 组成,堆叠的最高高度为 5 。
因此,返回 [2, 5, 5] 作为答案。

示例 2:

输入:positions = [[100,100],[200,100]]
输出:[100,100]
解释:
第 1 个方块掉落后,最高的堆叠由方块 1 组成,堆叠的最高高度为 100 。
第 2 个方块掉落后,最高的堆叠可以由方块 1 组成也可以由方块 2 组成,堆叠的最高高度为 100 。
因此,返回 [100, 100] 作为答案。
注意,方块 2 擦过方块 1 的右侧边,但不会算作在方块 1 上着陆。

在算法40中,我们介绍了线段树以及使用线段树求累加和的案例。他们使用的都是一维数据,区间很好划分。而这一题是二维数组,二维数组中的每个一维数组第一个元素代表起始位置,第二个元素代表高度,思来想去,还是不知道如何去划分区间。

1. 之前的区间跟数据没关系,只是跟数据的位置有关系;而本题尝试以二维数组作为一个单元数据,没法划分;

2. 以X轴横坐标划分区间;假设数据量很大,而且数据范围也很大,比如{{1,10}, {10,1000000},{100000000, 100000000000000000}}; 这样的数组该如何去划分区间呢?貌似也走不通。

离散化技巧:

假设二维数组为 {

{300, 5000},{17, 67300},{4500, 5000万}

}

我们把这些坐标进行搜集并排序得到 17,300,4500,5000,67300,5000万; 按照线段树的思路给下标:

下标 :0123456
忽略173004500500067300500万

如果按照这样的思路,我们就可以得到:

{300, 5000} = [2, 4] 区间

{17, 67300} = [1, 5] 区间

{4500, 5000万} = [3, 6] 区间

这样的技巧就叫做离散化技巧,确实很牛逼。我也是思考了很久,最终看了大神的解释才弄懂的。

本题分析;

假设二维数组中有一个数组 {1,3},1代表开始位置,3代表长度;得到开始、结束位置{1,4};

此时,又来一个数组 {4,2};代表4是开始位置,2是长度;得到开始、结束位置{4,6}

那么此时问题就来了,{1,4} 和 {4,6}存在相同的4;但是本题方块却可以正常紧挨着降落在一排;如果按照区间来划分算高度,4这个位置会出问题:

想要解决这样的问题,结束的位置坐标往左推一个就可以解决;

{1,4} 实际上代表的是 [1,4) 左闭右开; 给转换成 [1,3]

{4,6}实际上代表的是 [4,6) 左闭右开;给转换成 [4,5]

本题中的坐标都只能是整数,这是隐藏信息;因此,以上的转换是正确的。

区间的确定:

本题中,区间是根这些坐标有关系的;利用离散化技巧以及上方关于区间的分析可得;

我们以本题中给定的图片进行分析

第一个数是[1,3),我们得到[1,2]

第二个数是[2,5), 我们得到[2,4]

第三个数是[6,7), 我们得到[6,6]

无重复收集这些坐标信息,得到 {1,2,4,6},分别给个下标

下标01234
忽略1246

这样,我们就知道了第一个方块在 1-2区间;第二个方块在 2-3区间;第三个方块在4区间;

本题是算最大高度的,因此无需原始数组;max数组全部为0即可;

当第一个方块{1,3}落下的时候,{1,3} 对应 1-2区间,也就是跟节点的左子树;那么左子树的高度就为 3; 跟节点取左、右子节点的最大值;

第二个方块{2,5}落下,{2,5}对应 2-3区间;此时,1-2区间的3下方到左、右子节点;

获取到2-3区间的最大值;目测是3;那么此时第二块方块落下以后,2-3区间的最大值就为 3+2 = 5;根节点取左、右子树最大值,也为5;

第三个方块{6,7}对应4区间;那么4区间高度就为1; 取值结果没有变化

目测整个线段树的最大高度会一直汇总到树的顶部,那么只要获取树的顶部数据,就可以获取到最大值了;

package code04.线段树_02;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.TreeSet;

/**
 * 力扣 699题 :掉落的方块
 * https://leetcode.cn/problems/falling-squares/description/
 */
public class Code02_FallingSquares {

    class SegmentTree {
        int[] max;
        int[] lazy;
        boolean[] update;

        public SegmentTree(int size) {
            max = new int[size * 4];
            lazy = new int[size * 4];
            update = new boolean[size * 4];
        }

        //统计index节点,从左、右节点中选取较大的
        public void count(int index) {
            max[index] = Math.max(max[index * 2], max[index * 2 + 1]);
        }

        public void pushDown(int curIndex)
        {
            //判断curIndex是否有懒数据没更新到子节点
            if (lazy[curIndex] != 0) {
                //左、右子区间加上懒更新的数据
                max[curIndex * 2] = lazy[curIndex];
                max[curIndex * 2 + 1] =  lazy[curIndex];

                //左、右子树区间记录懒数据
                lazy[curIndex * 2] = lazy[curIndex];
                lazy[curIndex * 2 + 1] =  lazy[curIndex];

                //原curIndex数据已经下放到子区间了,此处需要重置为0
                lazy[curIndex] = 0;
            }
        }


        public void add(int left, int right, int curIndex, int start, int end, int value)
        {
            if (start <= left && end >= right) {
                //默认max中全部为高度全部为0. 那么下降一个方块,高度就选取大的加上下架的方块高度 value
                max[curIndex] = value;
                lazy[curIndex] = value;
                return;
            }

            int mid = (left + right)/2;
            //如果当前节点curIndex之前有懒的数据,那么把curIndex之前的懒
            //数据下放到子节点区间
            pushDown(curIndex);

            if (start <= mid) {
                add(left, mid, curIndex * 2, start, end, value);
            }

            if (end > mid) {
                add(mid + 1, right, curIndex * 2 + 1, start, end, value);
            }

            //重新汇总curIndex节点的最大值
            count(curIndex);
        }

        public int query(int left, int right, int curIndex, int start, int end)
        {
            if (start <= left && end >= right) {
                return max[curIndex];
            }

            int max = 0;
            int mid = (left + right) / 2;
            pushDown(curIndex);
            if (start <= mid) {
                int ans = query(left, mid, curIndex * 2, start, end);
                max = Math.max(ans, max);
            }

            if (end > mid) {
                int ans = query(mid + 1, right, curIndex * 2 + 1, start, end);
                max = Math.max(ans, max);
            }
            return max;
        }

        public int getTreeMaxHeight(){
            return max[1];
        }
    }


    public HashMap<Integer, Integer> index(int[][] positions)
    {
        TreeSet<Integer> pos = new TreeSet<>();
        //离散化过程,统计开始、结束区间的坐标。
        //不管数组长度为多少,最终都是落在这些区间中的
        for (int[] arr : positions) {
            pos.add(arr[0]);
            pos.add(arr[0] + arr[1] - 1);
        }

        int index = 1;
        HashMap<Integer, Integer> map = new HashMap<>();
        //给每个下标编个index,从1开始; 模拟原始线段树的原始数组中给每个元素添加下标的逻辑
       /* for(Iterator iterator = pos.iterator(); iterator.hasNext();) {
            int key = (int) iterator.next();
            map.put(key, index++);
        }*/
        for (Integer key : pos) {
            map.put(key, index++);
        }
        return map;
    }

    public List<Integer> fallingSquares(int[][] positions)
    {
        //获取到了X轴上对应的下标
        HashMap<Integer, Integer> map = index(positions);
        int size = map.size();
        SegmentTree segmentTree = new SegmentTree(size);

        int left = 1;
        int right = size;
        int curIndex = 1;

        List<Integer> list = new ArrayList<>();

        for (int i = 0; i < positions.length; i++) {
            //任务开始下标
            int startIndex = map.get(positions[i][0]);
            //修改的值
            int value = positions[i][1];
            //任务结束下标; 此下标代表 [1,4) 替换成 [1,3]
            int endIndex = map.get(positions[i][0] + value - 1);

            //这个地方比较有意思,假如3-4区域高度max为5.
            //再降落一块高度为3的石头在1-3区间。不考虑重力因素
            //1-3的高度应该为 5 + 3 = 8; 哪怕之前1-2区域高度为0
            int ans = segmentTree.query(left, right, curIndex, startIndex, endIndex);
            int height = ans + value;
            //降落一块方块
            segmentTree.add(left, right, curIndex, startIndex, endIndex, height);
            //全区间查找最大值
            System.out.println(segmentTree.getTreeMaxHeight());
            list.add(segmentTree.getTreeMaxHeight());
        }
        return list;
    }

    public static void main(String[] args) {
        Code02_FallingSquares ss = new Code02_FallingSquares();
        //输出2 5 5
        int[][] positions = {{1,2},{2,3},{6,1}};
        ss.fallingSquares(positions);
/*
        int[][] positions2 = {{100,100},{200,100}};
        ss.fallingSquares(positions2);

        int[][] positions3 = {{9,7},{1,9},{3,1}};
        ss.fallingSquares(positions3);

        int[][] positions4 = {{6,4},{2,7},{6,9}};
        ss.fallingSquares(positions4);*/
    }
}

一顿操作猛如虎,结果测试发现胜率不到10%;

查了好久代码,没有发现什么结构上的问题。最终只注释掉了一行:

System.out.println(segmentTree.getTreeMaxHeight());

再测试发现, 96%的胜率:

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1429245.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

CSS是一门需要单独学习的技术吗?

CSS (Cascading Style Sheets) &#xff0c;做前端开发的人都很清楚&#xff0c;因为这是他们的一项必不可少的技能。我以前也是知道CSS&#xff0c;但从来没有单独学习过&#xff0c;认为就它只是用来渲染网页的表现层效果&#xff0c;定制页面和内元素的布局、颜色和字体等&a…

学习Android的第二天

目录 Android User Interface 用户界面 UI Android View与ViewGroup的概念 Android View android.view.View android.view.View XML 属性 android:id 属性 Android ViewGroup android.view.ViewGroup ViewGroup.LayoutParams ViewGroup.MarginLayoutParams ViewGr…

深度学习(12)--Mnist分类任务

一.Mnist分类任务流程详解 1.1.引入数据集 Mnist数据集是官方的数据集&#xff0c;比较特殊&#xff0c;可以直接通过%matplotlib inline自动下载&#xff0c;博主此处已经完成下载&#xff0c;从本地文件中引入数据集。 设置数据路径 from pathlib import Path# 设置数据路…

Pytest框架测试

Pytest 是什么? pytest 能够支持简单的单元测试和复杂的功能测试;pytest 可以结合 Requests 实现接口测试; 结合 Selenium、Appium 实现自动化功能测试;使用 pytest 结合 Allure 集成到 Jenkins 中可以实现持续集成。pytest 支持 315 种以上的插件;为什么要选择 Pytest 丰…

2024年第九届信号与图像处理国际会议(ICSIP 2024)

2024第九届信号与图像处理国际会议&#xff08;ICSIP 2024&#xff09;将于2024年7月12-14日在中国南京召开。ICSIP每年召开一次&#xff0c;在过去的七年中吸引了1200多名与会者&#xff0c;是展示信号和图像处理领域最新进展的领先国际会议之一。本次将汇集来自亚太国家、北美…

基于SpringBoot+Vue的师生疫情健康信息管理登记平台,附源码

博主介绍&#xff1a;✌程序员徐师兄、7年大厂程序员经历。全网粉丝12w、csdn博客专家、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java技术领域和毕业项目实战✌ &#x1f345;文末获取源码联系&#x1f345; &#x1f447;&#x1f3fb; 精彩专栏推荐订阅&#x1f447;…

python+PyQt5 左右声道测试

UI&#xff1a; 源代码&#xff1a; # -*- coding: utf-8 -*-# Form implementation generated from reading ui file MicrophoneWinFrm.ui # # Created by: PyQt5 UI code generator 5.15.2 # # WARNING: Any manual changes made to this file will be lost when pyuic5 is …

Java正则表达式之Pattern和Matcher

目录 前言一、Pattern和Matcher的简单使用二、Pattern详解2.1 Pattern 常用方法2.1.1 compile(String regex)2.1.2 matches(String regex, CharSequence input)2.1.3 split(CharSequence input)2.1.4 pattern()2.1.5 matcher(CharSequence input) 三、Matcher详解3.1 Matcher 常…

​(三)hadoop之hive的搭建1

下载 访问官方网站https://hive.apache.org/ 点击downloads 点击Download a release now! 点击https://dlcdn.apache.org/hive/ 选择最新的稳定版 复制最新的url 在linux执行下载命令 wget https://dlcdn.apache.org/hive/hive-3.1.3/apache-hive-3.1.3-bin.tar.gz 2.解压…

账簿和明细账

目录 一. 账簿的意义和种类二. 明细账 \quad 一. 账簿的意义和种类 \quad 账簿是由一定格式、互有联系的账页组成&#xff0c;以审核无误的会计凭证为依据,用来序时地、分类地记录和反映各项经济业务的会计簿籍&#xff08;或称账本&#xff09;。设置和登记账簿是会计工作的重…

10个常考的前端手写题,你全都会吗?(上)

前言 &#x1f4eb; 大家好&#xff0c;我是南木元元&#xff0c;热爱技术和分享&#xff0c;欢迎大家交流&#xff0c;一起学习进步&#xff01; &#x1f345; 个人主页&#xff1a;南木元元 今天来分享一下10个常见的JavaScript手写功能。 目录 1.实现new 2.call、apply、…

C++:输入流/输出流

C流类库简介 C为了克服C语言中的scanf和printf存在的缺点。&#xff0c;使用cin/cout控制输入/输出。 cin&#xff1a;表示标准输入的istream类对象&#xff0c;cin从终端读入数据。cout&#xff1a;表示标准输出的ostream类对象&#xff0c;cout向终端写数据。cerr&#xff…

刘知远团队大模型技术与交叉应用L6-基于大模型文本理解和生成介绍

介绍 NLP的下游运用可以分为&#xff1a;NLU(理解)和NLG(生成) 信息检索&#xff1a;NLU 文本生成&#xff1a;NLG 机器问答&#xff1a;NLUNLG 大模型在信息检索 大模型在机器问答 大模型在文本生成 信息检索-Information Retrieval (IR) 背景 谷歌搜索引擎目前同时集成了…

水表表盘读数识别新体验,带你进入华为云ModelArts算法套件的世界

前言 数字时代&#xff0c;数字化服务已经发展到各行各业&#xff0c;我们的生活中也随处可见。 数字化服务的便捷了我们的衣食住行的方方面面&#xff0c;除了我们日常化的出行、饮食、购物&#xff0c;其实住方面也已有了很深的发展。 水电燃气这三项和我们生活息息相关的…

YOLO-World: Real-Time Open-Vocabulary Object Detection

文章目录 1. Introduction2. Experiments2.1 Implementation Details2.2 Pre-training2.3 Ablation Experiments2.3.1 预训练数据2.3.2 对RepVL-PAN的消融研究2.3.3 文本编码器 2.4 Fine-tuning YOLO-World2.5 Open-Vocabulary Instance Segmentation2.6 Visualizations Refere…

R语言分析任务:

有需要实验报告的可CSDN 主页个人私信 《大数据统计分析软件&#xff08;R语言&#xff09;》 实 验 报 告 指导教师&#xff1a; 专 业&#xff1a; 班 级&#xff1a; 姓 名&#xff1a; 学 …

Multisim14.0仿真(四十三)LM311应用

一、LM311简介&#xff1a; lm311是一款高灵活性的电压比较器&#xff0c;能工作于5V-30V单个电源或正负15V分离电源。 二、LM311主要特性&#xff1a; ★ 快速响应时间&#xff1a;165 ns。 ★ 选通能力。 ★ 最大输入偏置电流&#xff1a;300nA。 ★ 最大输入偏置电流&#…

8868体育助力意甲尤文图斯俱乐部 帮助球队签订新合同

意甲的尤文图斯俱乐部是8868合作体育球队之一&#xff0c;根据意大利媒体的消息&#xff0c;尤文图斯已经决定和费德里科-基耶萨续约&#xff0c;这名球员已经开始思考他的将来了。 费德里科-基耶萨今年26岁&#xff0c;他和尤文图斯的合约到2025年6月30号就结束了。他知道很多…

大数据分析|设计大数据分析的三个阶段

文献来源&#xff1a;Saggi M K, Jain S. A survey towards an integration of big data analytics to big insights for value-creation[J]. Information Processing & Management, 2018, 54(5): 758-790. 下载链接&#xff1a;链接&#xff1a;https://pan.baidu.com/s/1…

flask基于django大数据的证券股票分析系统python可视化大屏

证券分析系统采用B/S架构&#xff0c;数据库是MySQL。网站的搭建与开发采用了先进的Python进行编写&#xff0c;使用了Django框架。该系统从两个对象&#xff1a;由管理员和用户来对系统进行设计构建。主要功能包括&#xff1a;个人信息修改&#xff0c;对股票信息、股票买入、…