LeetCode每日一题 (28) 117. 填充每个节点的下一个右侧节点指针 II

战胜.

于 2020-09-28 22:57:49 发布

阅读量136

点赞数

分类专栏：战胜LeetCode刷题文章标签： leetcode

本文链接：https://blog.csdn.net/qq_45021180/article/details/108859328

版权

战胜LeetCode刷题专栏收录该内容

63 篇文章 0 订阅

订阅专栏

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

在这里插入图片描述

题解：

/*
// 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:
    // 通过层次遍历，使用队列，每一层做一次添加next操作
    Node* connect(Node* root) {
        queue<Node*> Q;
        Node *p,*pre;
        if(root==NULL) return root;
        Q.push(root);
        while(!Q.empty()){
            int len=Q.size();
            pre=Q.front();
            Q.pop();
            if(pre->left!=NULL) Q.push(pre->left);
            if(pre->right!=NULL) Q.push(pre->right);
            for(int i=1;i<len;i++){
                p=Q.front();
                Q.pop();
                pre->next=p;
                pre=p;
                if(p->left!=NULL) Q.push(p->left);
                if(p->right!=NULL) Q.push(p->right);
            }
            pre->next=NULL;
        }
        return root;
    }
};