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

最新推荐文章于 2021-08-30 20:44:43 发布

小白吖丶

最新推荐文章于 2021-08-30 20:44:43 发布

阅读量128

点赞数 1

分类专栏： leetcode 文章标签：二叉树层次遍历节点连接队列算法

本文链接：https://blog.csdn.net/qq_43180487/article/details/108856033

版权

leetcode 专栏收录该内容

170 篇文章 0 订阅

订阅专栏

/*
// 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) {
		queue<Node*> q;
		if (root != NULL) 
			q.push(root);
		while (!q.empty()) 
		{
			int size = q.size();
			Node *node;
			Node *pre_node;
			for (int i = 0; i < size; i++)
			{
				if (i == 0)
				{
					node = q.front();
					q.pop();
					pre_node = node;
				}
				else
				{
					node = q.front();
					q.pop();
					pre_node->next = node;
					pre_node = pre_node->next;
				}
                if(node->left)
                {
                    q.push(node->left);
                }
                if(node->right)
                {
                    q.push(node->right);
                }
			}
			pre_node->next == NULL;
		}
		return root;

	}
};