leetcode Populating Next Right Pointers in Each Node &&Populating Next Right Pointers in Each Node||

Do_Cool_Thing

于 2014-08-15 20:16:21 发布

阅读量277

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode Fighting

本文链接：https://blog.youkuaiyun.com/menyangyang/article/details/38589639

leetcode Fighting 专栏收录该内容

133 篇文章

订阅专栏

本文介绍了一种用于优化树节点链接的算法，通过队列操作和指针修改实现节点之间的连接，空间复杂度为O(n^2)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

两题菊科使用下面的算法，不过此算法的空间复杂度为O(n^2).

代码

class Solution
{
public:
	void connect(TreeLinkNode *root)
	{
		TreeLinkNode *pre, *tempNode;
		queue<TreeLinkNode*>queue_pop, queue_push;	
		bool flag = true;

		if(root==NULL)
			return ;

		queue_pop.push(root);

		while(!queue_pop.empty())
		{	
			if(queue_push.empty()&&flag)
			{
				flag = false;
				tempNode = queue_pop.front();
				pre = tempNode;
			}
			else
			{
				tempNode = queue_pop.front();
				pre ->next= tempNode;
				pre = tempNode;
			}

			queue_pop.pop();

			if(queue_pop.empty())
			{
				tempNode->next = NULL;
			}

			if(tempNode->left)
				queue_push.push(tempNode->left);
			if(tempNode->right)
				queue_push.push(tempNode->right);
			
			if(queue_pop.empty())
			{
				flag = true;
				swap(queue_pop, queue_push);
			}
		}
	
	
	}

};