JAVA学习-练习试用Java实现“填充每个节点的下一个右侧节点指针”

问题：

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

struct Node {

int val;

Node *left;

Node *right;

Node *next;

}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

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

进阶：

只能使用常量级额外空间。

使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

示例：

输入：root = [1,2,3,4,5,6,7]

输出：[1,#,2,3,#,4,5,6,7,#]

解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。序列化的输出按层序遍历排列，同一层节点由 next 指针连接，'#' 标志着每一层的结束。

提示：

树中节点的数量少于 4096

-1000 <= node.val <= 1000

解答思路：

以下是使用 Java 语言实现填充每个节点的下一个右侧节点指针的代码：

class Node {

    int val;

    Node left;

    Node right;

    Node next;



    Node(int val) {

        this.val = val;

    }

}



public class ConnectNodes {



    public Node connect(Node root) {

        if (root == null) {

            return null;

        }



        // 进行层序遍历

        Queue<Node> queue = new LinkedList<>();

        queue.add(root);



        while (!queue.isEmpty()) {

            int size = queue.size();



            for (int i = 0; i < size; i++) {

                Node current = queue.poll();



                // 填充 next 指针

                if (i < size - 1) {

                    current.next = queue.peek();

                }



                // 加入左右子节点

                if (current.left!= null) {

                    queue.add(current.left);

                }

                if (current.right!= null) {

                    queue.add(current.right);

                }

            }

        }



        return root;

    }



    public static void main(String[] args) {

        Node root = new Node(1);

        root.left = new Node(2);

        root.right = new Node(3);

        root.left.left = new Node(4);

        root.left.right = new Node(5);

        root.right.left = new Node(6);

        root.right.right = new Node(7);



        ConnectNodes cn = new ConnectNodes();

        Node connectedRoot = cn.connect(root);



        // 打印结果

        Queue<Node> queue = new LinkedList<>();

        queue.add(connectedRoot);



        while (!queue.isEmpty()) {

            Node current = queue.poll();

            System.out.print(current.val + " ");

            if (current.next!= null) {

                System.out.print("-> ");

            }

            if (current.left!= null) {

                queue.add(current.left);

            }

            if (current.right!= null) {

                queue.add(current.right);

            }

        }

    }

}

在上述代码中，我们定义了一个'Node'类来表示二叉树的节点。'connect'方法用于填充每个节点的'next'指针。

- 首先，判断根节点是否为空，如果为空则直接返回。

- 然后，创建一个队列，并将根节点加入队列。

- 进入一个循环，每次循环取出队列头部的节点。

- 填充当前节点的'next'指针，如果不是当前层的最后一个节点，则将其'next'指针指向队列中的下一个节点。

- 将当前节点的左右子节点加入队列。

- 循环结束后，返回根节点。

在'main'方法中，我们创建了一个示例二叉树，并调用'connect'方法进行连接。最后，通过层序遍历打印出连接后的结果。

(文章为作者在学习java过程中的一些个人体会总结和借鉴，如有不当、错误的地方，请各位大佬批评指正，定当努力改正，如有侵权请联系作者删帖。)