LeetCode116. 填充每个节点的下一个右侧节点指针

LeetCode116. 填充每个节点的下一个右侧节点指针

🍕🎉🥓🎊🚗🐷😱❤😀❣🥙🎇🕵️‍♀️😎🎑🎁🎉😚🎇

---

题目：

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

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,\#]$
• 解释： 给定二叉树如左图所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如右图所示。序列化的输出按层序遍历排列，同一层节点由 next 指针连接，'#' 标志着每一层的结束。

解题思路：

 利用队列，每遍历一个元素，就将其加入到队列中，判断当前元素是否为该层最后一个元素，若是的话则将其next置为空，否则next指向下一个元素。

C++版本

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL) return NULL;
        queue<Node*> q;
        q.push(root);
        while(!q.empty()){
            int size = q.size();
            for(int i = 0; i < size; i++){
                Node* temp = q.front();
                q.pop();
                if(i != size - 1) temp->next = q.front();
                else temp->next = NULL;
                if(temp->left) q.push(temp->left);
                if(temp->right) q.push(temp->right);
            }
        }
        return root;
    }
};

Java版本

class Solution {
    public Node connect(Node root) {
        Queue<Node> q = new LinkedList<>();
        if(root == null) return null;
        q.offer(root);
        while(!q.isEmpty()){
            int size = q.size();
            for(int i = 0; i < size; i++){
                Node temp = q.poll();
                if(i == size - 1) temp.next = null;
                else temp.next = q.peek();
                if(temp.left != null) q.offer(temp.left);
                if(temp.right != null) q.offer(temp.right);
            }
        }
        return root;
    }
}