算法题之填充每个节点的下一个右侧节点指针 II

算法题之填充每个节点的下一个右侧节点指针 II

题目描述

题解

java代码

public class FillUpRightNode {
    /*
    *给定一个二叉树:填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。
    * 输入：root = [1,2,3,4,5,null,7]
    * 输出：[1,#,2,3,#,4,5,7,#]
     * */
    public static void main(String[] args) {
        Node root = new Node(1);
        Node node2 = new Node(2);
        Node node3 = new Node(3);
        Node node4 = new Node(4);
        Node node5 = new Node(5);
        Node node6 = new Node(7);
        root.left = node2;
        root.right = node3;
        node2.left = node4;
        node2.right = node5;
        node3.right = node6;
        FillUpRightNode solution = new FillUpRightNode();
        solution.connect(root);
        //测试
        System.out.println(node6.next);
    }

    public Node connect(Node root) {
        //判断root是否为null
        if (root == null) {
            //该二叉树为空
            return null;
        }
        //将root节点赋值给cur,开始进行BFS遍历
        Node cur = root;
        //cur:取整个二叉树的每一层的最左侧节点,由于第一层只有一个root节点,且next默认初始化为null,故第一层不用进行BFS遍历,创建链表设置其next指针
        while (cur != null) {
            //定义一个链表的头节点:只是作为BFS每层链表的头结点的标志位
            Node head = new Node(0);
            //定义pre作为每一层链表节点的前一个节点
            Node pre = head;
            //进行当前层的BFS遍历
            while (cur != null) {
                //先判断左子节点是否存在,若存在,则加入链表并将链表的pre指针指向下一个节点
                if (cur.left != null) {
                    pre.next = cur.left;
                    pre = pre.next;
                }
                //再判断右子节点是否存在,若存在,则加入链表并将链表的pre指针指向下一个节点
                if (cur.right != null) {
                    pre.next = cur.right;
                    pre.next = pre;
                }
                //将cur指向此层的next节点,继续进行链表的设置
                cur =cur.next;
            }
            //当前层BFS遍历结束,将cur指向head下一个节点,即下一层的最左侧子节点,进行下一轮的BFS
            //直到cur为null,即遍历到最左侧子节点为null(此二叉树遍历结束)停止
            cur = head.next;
        }
        //返回填充完next指针以后的根节点
        return root;
    }
}

//定义树的节点类
class Node {
    public int val;
    public Node left;
    public Node right;
    public Node next;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, Node _left, Node _right, Node _next) {
        val = _val;
        left = _left;
        right = _right;
        next = _next;
    }

    @Override
    public String toString() {
        return "Node{" +
                "val=" + val +
                '}';
    }
}