填充每个节点的下一个右侧节点指针

填充每个节点的下一个右侧节点指针

给定一个 完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL

初始状态下,所有 next 指针都被设置为 NULL

示例 1:

输入:root = [1,2,3,4,5,6,7]
输出:[1,#,2,3,#,4,5,6,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化的输出按层序遍历排列,同一层节点由 next 指针连接,'#' 标志着每一层的结束。

方法一:层次遍历

public Node connect(Node root) {
    if (root == null) {
        return root;
    }
    Node res = root;
    Queue<Node> queue = new LinkedList<>();
    queue.offer(root);
    while (!queue.isEmpty()) {
        int size = queue.size();
        for (int i = 0; i < size; i++) {
            root = queue.poll();
            if (i < size - 1) {
                root.next = queue.peek();
            }
            if (root.left != null) {
                queue.offer(root.left);
            }
            if (root.right != null) {
                queue.offer(root.right);
            }
        }
    }
    return res;
}

方法二:使用已建立的next指针
第一种 是这两个串联的节点都有一个共同的父节点,通过父节点就可以将这两个子节点串联起来


roo.left.next => root.right

第二种 是这两个串联的节点的父节点不同,对于这种情况,如果我们能将这一层的上一层串联好。那么可以通过父节点的next找到邻居,完成串联。


root.right.next => root.next.left
这里我们需要保证 root.next 不为空就可以了。

也就是说当我们要串联第 i 层节点时,需要先完成第 i-1 层的节点串联
第一层最多只有一个节点,不需要串联
第二层最多只有两个节点,借助根节点就可以完成串联了
第三层串联时,上一层已经串联完了,所以第三层可以完成串联
同理,可以完成第四层,第五层,第N层的串联

public Node connect(Node root) {
    if (root == null) {
        return root;
    }
    Node head = root;
    while (head.left != null) {
        Node p = head;
        while (p != null) {
            p.left.next = p.right;
            if (p.next != null) {
                p.right.next = p.next.left;
            }
            p = p.next;
        }
        head = head.left;    
    }
    return root;
}

方法三:递归

public Node connect(Node root) {
    if (root == null) {
        return null;
    }
    if (root.left != null) {
        root.left.next = root.right;
        if (root.next != null) {
            root.right.next = root.next.left;
        }
        connect(root.left);
        connect(root.right);
    }
    return root;
}

填充每个节点的下一个右侧节点指针 II

给定一个二叉树:

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL
初始状态下,所有 next 指针都被设置为 NULL

示例 1

输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化输出按层序遍历顺序(由 next 指针连接),'#' 表示每层的末尾。

使用已建立的 next 指针

  • 从第一层开始(第一层只有一个 root 节点),每次循环遍历当前层的链表节点,通过节点的 left 和 right 得到下一层的节点。
  • 把下一层的节点从左到右连接成一个链表。
  • 拿到下一层链表的头节点,进入下一轮循环。

代码实现时,可以用一个哨兵节点来表示「第一个节点之前的节点」,从而减少一些关于空节点的判断逻辑。

public Node connect(Node root) {
    if (root == null) {
        return root;
    }
    Node dummy = new Node();
    Node p = root;
    while (p != null) {
        Node q = dummy;
        while (p != null) {
            if (p.left != null) {
                q.next = p.left;
                q = q.next;
            }
            if (p.right != null) {
                q.next = p.right;
                q = q.next;
            }
            p = p.next;
        }
        p = dummy.next;
        dummy.next = null;
    }
    return root;
}

递归
DFS 这棵树,从根节点 1 出发,向左递归到 2,再向左递归到 4。
这三个节点正好是每一层的第一个节点(类似链表头),用一个数组 pre 记录,即 pre[0] 为节点 1,pre[1] 为节点 2,pre[2] 为节点 4。pre的下标就是节点的深度。
继续递归到 5(深度为 2),从 pre[2] 中拿到节点 4,把 4 的 next 指向 5。然后更新 pre[2] 为节点 5,这样在后面递归到节点 7 时,就可以从 pre[2] 中拿到节点 5,把 5 的 next 指向 7 了。

private List<Node> pre = new ArrayList<>();

public Node connect(Node root) {
    dfs(root, 0);
    return root;
}

private void dfs(Node root, int level) {
    if (root == null) {
        return;
    }
    if (level == pre.size()) {
        pre.add(root);
    } else {
        pre.get(level).next = root;
        pre.set(level, root);
    }
    dfs(root.left, level + 1);
    dfs(root.right, level + 1);
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,324评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,303评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,192评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,555评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,569评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,566评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,927评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,583评论 0 257
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,827评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,590评论 2 320
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,669评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,365评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,941评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,928评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,159评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,880评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,399评论 2 342

推荐阅读更多精彩内容