Java 算法-线段树的修改

  首先说明一下,看这个博客之前,最好有线段树的基本概念,比如说线段树的构造、线段树的查询之类的。
最近在学习ANdroid IPC,所以有几天没在LintCode上面做面试题,今天晚上恰巧有空,做了一道题。线段树在这之前,我认真的学习了一遍,其实我也是初学线段树,之前在别人的博客上学习。这里我就不在详细的讲解线段树,因为网络上那些大佬比我讲的更加清楚,我在推荐一下一个大佬的博客:浅谈线段树。我就是在这个博客上学习的。
  废话说的有点多,直接进入今天的主题。

题意:
对于一棵 最大线段树, 每个节点包含一个额外的 max 属性,用于存储该节点所
代表区间的最大值。

设计一个 modify 的方法,接受三个参数 root、 index 和 value。该方法将 
root 为根的线段树中 [start, end] = [index, index] 的节点修改为了新的 
value ,并确保在修改后,线段树的每个节点的 max 属性仍然具有正确的值。
样例:
对于线段树:

                      [1, 4, max=3]
                    /                \
        [1, 2, max=2]                [3, 4, max=3]
       /              \             /             \
[1, 1, max=2], [2, 2, max=1], [3, 3, max=0], [4, 4, max=3]
如果调用 modify(root, 2, 4), 返回:

                      [1, 4, max=4]
                    /                \
        [1, 2, max=4]                [3, 4, max=3]
       /              \             /             \
[1, 1, max=2], [2, 2, max=4], [3, 3, max=0], [4, 4, max=3]

或 调用 modify(root, 4, 0), 返回:

                      [1, 4, max=2]
                    /                \
        [1, 2, max=2]                [3, 4, max=0]
       /              \             /             \
[1, 1, max=2], [2, 2, max=1], [3, 3, max=0], [4, 4, max=0]

  乍一看,这个题不是非常的简单吗?是啊,非常的简单,使用递归的方式是非常的简单的。

1.递归

  递归的方式非常的简单,这里就不在讲解,直接看代码

public void modify(SegmentTreeNode root, int index, int value) {
        if (index > root.end || index < root.start) {
            return;
        }
        //当找到那个那个点了,更新max值
        if (root.start == root.end && root.start == index) {
            root.max = value;
            return;
        }
        int mid = (root.start + root.end) / 2;
        //遍历左子树
        if (index <= mid) {
            modify(root.left, index, value);
        } else {  //遍历右子树
            modify(root.right, index, value);
        }
        //更新root节点本身,这一点非常的重要,因为子树的max值改了
        //root的max有可能会改变
        root.max = Math.max(root.left.max, root.right.max);
    }

2.非递归的方式(o(n)的时间复杂度)

  有人有疑问,为什么递归的方式都能简单的解决问题,为什么我们偏要将问题想得非常的复杂。其实,这个题有一个挑战项,就是在O(n)的复杂度里面解决这种问题。
  遇到这种需求,才开始,我也是一脸懵逼的,但是经过后来借鉴了别人的思想,能成功的理解,并且写出完整的代码。
  首先,我来解释这种方式的思路:

  (1).当value > root.max时,这种情况下,root的max肯定是value。因此这种情况,也是非常容易的解决的。使用循环遍历,更新每一个点的max值就行了。
  (2).当value <= root.max时,这种情况下,我们首先只能更新目标节点的max值,然后更新目标节点的上层节点,直到更新到root节点。感觉这里非常的麻烦,怎么更新上层节点的值呢?这里的方法相对来说比较巧妙。
public void modify(SegmentTreeNode root, int index, int value) {
        if (root == null || index > root.end || index < root.start) {
            return;
        }
        SegmentTreeNode node = root;
        //当value大于root.max,直接更新
        if (value > node.max) {
            while (node.left != null) {
                node.max = value;
                int mid = (node.end + node.start) / 2;
                if (index <= mid) {
                    node = node.left;
                } else {
                    node = node.right;
                }
            }
            node.max = value;
        } else {
            
            while (node.left != null) {
                int mid = (node.end + node.start) / 2;
                if (index <= mid) {
                    node = node.left;
                } else {
                    node = node.right;
                }
            }
            //先只更新目标节点的值
            node.max = value;
            
            SegmentTreeNode node1 = root;
            while (node != root) {
                node1 = root;
                //找到node节点的上一层节点
                while (node1.left != node && node1.right != node) {
                    int mid = (node1.end + node1.start) / 2;
                    if (index <= mid) {
                        node1 = node1.left;
                    } else {
                        node1 = node1.right;
                    }
                }
                //获得最大值
                int max = Math.max(node1.left.max, node1.right.max);
                if (max == node1.max) {
                    break;
                } else {
                    //更新max值,并且将node节点改变成node节点的上层节点
                    node1.max = max;
                    node = node1;
                }
            }
        }
    }
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,847评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,208评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,587评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,942评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,332评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,587评论 1 218
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,853评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,568评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,273评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,542评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,033评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,373评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,031评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,073评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,830评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,628评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,537评论 2 269

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,100评论 18 139
  • 参考两篇其他bolg总结的二叉树:https://github.com/xy7313/lintcode/blob/...
    暗黑破坏球嘿哈阅读 2,273评论 0 1
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 170,568评论 25 707
  • 树的概述 树是一种非常常用的数据结构,树与前面介绍的线性表,栈,队列等线性结构不同,树是一种非线性结构 1.树的定...
    Jack921阅读 4,373评论 1 31
  • 人物关系:小童老师,年龄17岁,是张同学的儿子。 在午餐的课堂上,小童老师,对只在意学习过程不追...
    張夏阅读 327评论 1 1