LC-填充每个节点指向最右节点的next指针II

最新推荐文章于 2025-08-13 22:08:56 发布

vector6_

最新推荐文章于 2025-08-13 22:08:56 发布

阅读量136

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签：数据结构 leetcode 二叉树指针

本文链接：https://blog.csdn.net/qq_41105501/article/details/109323506

Leetcode 专栏收录该内容

137 篇文章

订阅专栏

本文介绍了一种在二叉树中填充每个节点指向最右节点的next指针的算法。通过迭代方式实现，适用于任意二叉树结构，确保每个节点的next指针正确指向其最右侧的相邻节点。

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

填充每个节点指向最右节点的next指针II

//填充每个节点指向最右节点的next指针II
/*
继续思考"填充每个节点指向最右节点的next指针" 这道题
如果给定的树可以是任意的二叉树呢?你之前的给出的算法还有效吗?
*/
#include<vector>
using namespace std;
struct TreeLinkNode {
	int val;
	TreeLinkNode *left, *right, *next;
	TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
};
class Solution {
public:
	void connect(TreeLinkNode *root) {
		while (root)
		{
			TreeLinkNode dummy(-1);
			TreeLinkNode *prev;
			prev = &dummy;
			for (auto p = root; p; p = p->next)
			{
				if (p->left)
				{
					prev->next = p->left;
					prev = prev->next;
				}
				if (p->right)
				{
					prev->next = p->right;
					prev = prev->next;
				}
			}
			root = dummy.next;
		}
	}



	void Connect(TreeLinkNode* root)
	{
		if (!root)
			return;
		TreeLinkNode *p = root;
		TreeLinkNode *q;
		while (p->left)
		{
			q = p;
			while (q)
			{
				if (q->left && q->right)
				{
					q->left->next = q->right;
				}
				if (q->left && !q->right && q->next)
				{
					TreeLinkNode* q2 = q;
					while (q2->next && !q2->next->left && !q2->next->right)
					{
						q2 = q2->next;
					}
					if (q2->next && q2->next->left)
						q->left->next = q2->next->left;
					if (q2->next && q2->next->right)
						q->left->next = q2->next->right;
				}

				if (q->right && q->next)
				{
					TreeLinkNode *q3 = q;
					while (q3->next && !q3->next->left && !q3->next->right)
					{
						q3 = q3->next;
					}
					if (q3->next && q3->next->left)
						q->right->next = q3->next->left;
					if (q3->next && q3->next->right)
						q->right->next = q3->next->right;
				}
				q = q->next;
			}
			p = p->left;
		}
	}
};