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

最新推荐文章于 2023-05-22 16:24:08 发布

vector6_

最新推荐文章于 2023-05-22 16:24:08 发布

阅读量143

点赞数

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

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

版权

Leetcode 专栏收录该内容

137 篇文章 0 订阅

订阅专栏

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

//填充每个节点指向最右节点的next指针
/*
给定一个二叉树
struct TreeLinkNode {
TreeLinkNode *left;
TreeLinkNode *right;
TreeLinkNode *next;
}
填充所有节点的next指针，指向最接近它的同一层右边节点。
如果没有同一层没有右边的节点，则应该将next指针设置为NULL。
初始时，所有的next指针都为NULL
注意：
你只能使用常量级的额外内存空间
可以假设给出的二叉树是一个完美的二叉树(即，所有叶子节点都位于同一层，
而且每个父节点都有两个孩子节点)。 */
#include<vector>
using namespace std;
//Definition for binary tree with next pointer.
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) {
		if (root == nullptr)
			return;
		if (root->left && root->right)
			root->left->next = root->right;
		if (root->next && root->right)
			root->right->next = root->next->left;
		connect(root->left);
		connect(root->right);
	}

	//非递归 层序遍历
	//保存最左侧节点，建立next的同时，做层次遍历，直到null为止，
	//然后将变量跳到下一个阶层的最左节点，在边建立边层次
	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->next)
				{
					q->right->next = q->next->left;
				}
				q = q->next;
			}
			p = p->left;
		}
	}
};