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

问题：

给定一个二叉树

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

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

进阶：你只能使用常量级额外空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。
示例：

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

树中的节点数小于 6000
-100 <= node.val <= 100

解答思路：

以下是一个使用 Java 语言编写的解决方案，用于填充二叉树中每个节点的下一个右侧节点指针：

class Node {

    int val;

    Node left;

    Node right;

    Node next;



    Node(int val) {

        this.val = val;

    }

}



public class NextRightPointersInBinaryTree {

    public Node connect(Node root) {

        if (root == null) {

            return null;

        }



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

        queue.offer(root);



        while (!queue.isEmpty()) {

            int size = queue.size();

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

                Node node = queue.poll();



                if (node.left!= null) {

                    queue.offer(node.left);

                }

                if (node.right!= null) {

                    queue.offer(node.right);

                }



                if (i < size - 1) {

                    node.next = queue.peek();

                }

            }

        }



        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.right = new Node(7);



        NextRightPointersInBinaryTree solution = new NextRightPointersInBinaryTree();

        Node connectedRoot = solution.connect(root);



        // 打印连接后的二叉树

        printBinaryTree(connectedRoot);

    }



    public static void printBinaryTree(Node root) {

        if (root == null) {

            return;

        }



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

        queue.offer(root);



        while (!queue.isEmpty()) {

            int size = queue.size();

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

                Node node = queue.poll();

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



                if (node.left!= null) {

                    queue.offer(node.left);

                }

                if (node.right!= null) {

                    queue.offer(node.right);

                }

            }

            System.out.println();

        }

    }

}

这个解决方案使用了广度优先搜索（BFS）的方法。它使用一个队列来遍历二叉树的每一层，并在遍历过程中填充每个节点的 next 指针。具体来说，对于每一层，它将节点从队列中取出，将其左右子节点加入队列，并将当前节点的 next 指针设置为队列中的下一个节点（如果存在）。这样，通过逐层遍历，就可以正确地填充每个节点的 next 指针。

在 'main' 方法中，我们构建了一个示例二叉树，并调用 'connect' 方法填充 next 指针。最后，我们使用 'printBinaryTree' 方法打印连接后的二叉树。

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