LeetCode-116-填充每個節點的下一個右側節點指標

填充每個節點的下一個右側節點指標

題目描述：給定一個 完美二叉樹 ，其所有葉子節點都在同一層，每個父節點都有兩個子節點。二叉樹定義如下：

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

填充它的每個 next 指標，讓這個指標指向其下一個右側節點。如果找不到下一個右側節點，則將 next 指標設定為 NULL。

初始狀態下，所有 next 指標都被設定為 NULL。

示例說明請見LeetCode官網。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：層序遍歷

• 首先，如果root為空或者左右子節點都為空，則不需要處理next指標，直接返回root。
• 否則，當二叉樹不只有一個節點時，利用佇列對二叉樹進行層序遍歷記錄二叉樹每一層的節點，然後按順序處理當前層每一個節點的next指標。由於處理過程中所有的節點順序並沒有進行改變，所以最後返回root。

import com.kaesar.leetcode.Node;

import java.util.LinkedList;
import java.util.Queue;

public class LeetCode_116 {
    public static Node connect(Node root) {
        // 如果root為空或者左右節點都為空，不需要處理，直接返回root
        if (root == null) {
            return root;
        }
        if (root.left == null && root.right == null) {
            return root;
        }
        // 利用佇列記錄每層的節點
        Queue<Node> nodes = new LinkedList<>();
        nodes.add(root);
        while (!nodes.isEmpty()) {
            int count = nodes.size();
            Node last = nodes.poll();
            if (last.left != null) {
                nodes.add(last.left);
            }
            if (last.right != null) {
                nodes.add(last.right);
            }

            count--;
//             處理每層的節點的next指標
            while (count > 0) {
                Node curNode = nodes.poll();
                if (curNode.left != null) {
                    nodes.add(curNode.left);
                }
                if (curNode.right != null) {
                    nodes.add(curNode.right);
                }
                last.next = curNode;
                last = curNode;
                count--;
            }

        }
        return root;
    }

    public static void main(String[] args) {
        // 測試用例
        Node root = new Node(1);
        root.left = new Node(2);
        root.right = new Node(3);
        // 處理之前
        root.print();
        connect(root);
        System.out.println();
        // 處理之後
        root.print();
    }
}

【每日寄語】 好好學習，天天向上。