Day19力扣打卡

打卡记录

填充每个节点的下一个右侧节点指针 II（BFS层序遍历）

链接

采用BFS层序遍历，将每一层的节点遍历之后存入数组中，再对数组中一层中的所有节点遍历来进行next连接，再同时更新下一层的所有节点到数组当中。

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

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    Node* connect(Node* root) {
        if (!root) return nullptr;
        vector<Node*> q;
        q.push_back(root);
        while (!q.empty()) {
            vector<Node*> tmp;
            for (int i = 0; i < q.size(); ++i) {
                Node* node = q[i];
                if (i) q[i - 1]->next = node;
                if (node->left) tmp.push_back(node->left);
                if (node->right) tmp.push_back(node->right);
            }
            swap(q, tmp);
        }
        return root;
    }
};