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

填充每个节点的下一个右侧节点指针 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。

大致思路：

1. 通过层次遍历遍历二叉树
2. 将每层的结点入队，如果这个层次的队列为空，那么该结点的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;
    }
};

运行结果：

内存消耗比较大…
但是速度相比递归还是较快的