力扣116_填充每个节点的下一个右侧节点指针

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

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

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

示例:
在这里插入图片描述

算法思想:
有了前面翻转二叉树的经验,这里我们很容易通过观察发现,可以使root左孩子的next指向root的右孩子,然后递归的处理左右孩子,代码如下所示:

Node connect(Node root) {
    if (root == null || root.left == null) {
        return root;
    }

    root.left.next = root.right;

    connect(root.left);
    connect(root.right);

    return root;
}

但这里有一个问题,如图所示。
在这里插入图片描述
5,6节点之间的无法连接,因为我们的算法连接的是同一个父节点下的两个子节点,这显然是这道题的难点所在,如果只依赖一个节点(这里是父节点)的话,肯定是没办法连接「跨父节点」的两个相邻节点的。
我们的做法就是增加函数参数,一个节点做不到,我们就给他安排两个节点(这里是node1和node2),「将每一层二叉树节点连接起来」可以细化成「将每两个相邻节点都连接起来」:

class Solution {
    public Node connect(Node root) {
        if(root == null) return null;
        connectTwoNode(root.left, root.right);
        return root;
    }

    public void connectTwoNode(Node node1, Node node2) {
        if(node1 == null || node2 == null) return;
        //连接node1与node2
        node1.next = node2;
        //递归的连接node1的两个子节点
        connectTwoNode(node1.left, node1.right);
        //递归的连接node2的两个子节点
        connectTwoNode(node2.left, node2.right);
        //递归的连接node1的右节点与node2的左节点
        connectTwoNode(node1.right, node2.left);     
    }
}

复杂度分析:
后补。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值