题目描述:
给定一个 完全二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
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);
}
}
复杂度分析:
后补。