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

最新推荐文章于 2021-12-24 20:59:05 发布

一定要AC啊

最新推荐文章于 2021-12-24 20:59:05 发布

阅读量152

点赞数

分类专栏：数据结构文章标签： LeetCode

本文链接：https://blog.csdn.net/qq_43399971/article/details/103217627

版权

数据结构专栏收录该内容

10 篇文章 0 订阅

订阅专栏

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

LeetCode对应链接–https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii/

题目描述

给定一个二叉树

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。
在这里插入图片描述
大致思路：

通过层次遍历遍历二叉树
将每层的结点入队，如果这个层次的队列为空，那么该结点的next为空，否则为队首结点

代码实现

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL)
            return NULL;
        queue<Node *> q;
        
        q.push(root);
        while(!q.empty())
        {
        	queue<Node*> q0=q;
			while(!q0.empty())
			{ 	
				Node* p=q0.front();
				q0.pop();
				q.pop();
				if(q0.empty())
              		p->next = NULL;
           		else
               		p->next = q0.front();
				if (p->left != NULL)
					q.push(p->left);
				if (p->right != NULL)
					q.push(p->right);
			}
        }
        return root;
    }
};