LeetCode - 116. Populating Next Right Pointers in Each Node

题目

LeetCode - 116. Populating Next Right Pointers in Each Node

题目链接

https://leetcode.com/problems/populating-next-right-pointers-in-each-node/

参考博客

• [1] https://www.cnblogs.com/grandyang/p/4051326.html

解题思路

此题用深度优先遍历，这个容易想到，但是如何判断一个层的最后一个节点一开始没思路，此处参考了博客[1]。

总结:

深度优先遍历并判断同一层的第一个节点和最后一个节点的方法：

每次进入while循环，当前队列中存放的即为当前层的所有元素，用for循环依次弹出q.size()次，即可取得当前层的所有节点。

解题源码

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() {}

    Node(int _val, Node* _left, Node* _right, Node* _next) {
        val = _val;
        left = _left;
        right = _right;
        next = _next;
    }
};
*/
class Solution {
public:
    void bfs(Node* root){
        queue<Node*> q;
        q.push(root);
        Node* pre = nullptr;
        while(!q.empty()){
            pre = nullptr;
            for (int i = q.size() - 1; i >= 0; i--){
                
                Node* h = q.front();
                q.pop();

                if (pre){
                    pre->next = h;
                    pre = h;
                }else{
                    pre = h;
                }
                if (h && h->left){
                    q.push(h->left);
                    q.push(h->right);                
                }
            }
        }
    }
    Node* connect(Node* root) {
        bfs(root);
        return root;
    }
};