tree：二叉树next指针

最新推荐文章于 2024-02-07 21:11:28 发布

原创最新推荐文章于 2024-02-07 21:11:28 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

leet 专栏收录该内容

88 篇文章

订阅专栏

本文介绍了一个LeetCode上的算法问题——Populating Next Right Pointers in Each Node，通过C++实现将二叉树的每个节点与其右兄弟节点连接起来。主要涉及二叉树遍历和指针操作，时间复杂度为O(n)，空间复杂度为O(logn)。

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

Leetcode, PopulatingNextRightPointersInEachNode

#include <iostream>
#include <vector>
#include <stack>
#include <algorithm>
using namespace std;

struct TreeNode
{
	int _val;
	TreeNode *_left, *_right, *_next;
	TreeNode(int x) : _val(x), _left(nullptr), _right(nullptr), _next(nullptr) { }
};

void connect(TreeNode *root, TreeNode *sibling)
{
	if (root == nullptr) 
		return;
	else
		root->_next = sibling;

	connect(root->_left, root->_right );
	if (sibling)
		connect(root->_right, sibling->_left);
	else
		connect(root->_right, nullptr);
}

//时间复杂度O（n）,空间复杂度O（logn）
void solution(TreeNode *root)
{
	connect(root, nullptr);
}

int main()
{
	TreeNode *node1 = new TreeNode(1);
	TreeNode *node2 = new TreeNode(2);
	TreeNode *node3 = new TreeNode(3);
	TreeNode *node4 = new TreeNode(4);
	TreeNode *node5 = new TreeNode(5);
	TreeNode *node6 = new TreeNode(6);
	TreeNode *node7 = new TreeNode(7);
	node1->_left = node2;
	node1->_right = node3;
	node2->_left = node4;
	node2->_right = node5;
	node3->_left = node6;
	node3->_right = node7;

	solution(node1);

	return 0;
}