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

雄獅虎豹 發表於 2021-11-27
LeetCode

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

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

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();
    }
}
【每日寄語】 好好學習,天天向上。