[Leetcode] Flatten Binary Tree to Linked List

最新推荐文章于 2024-01-17 15:25:50 发布

原创最新推荐文章于 2024-01-17 15:25:50 发布 · 269 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

36 篇文章

订阅专栏

本文介绍如何通过前序和中序遍历的方式将二叉树转换为单链表和双向链表的方法。通过修改节点连接方式实现，适用于计算机科学中的数据结构转换。

题目：

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

The flattened tree should look like:

将二叉树转换为链表，注意到输出的顺序跟前序完全一样，所以自然想到在前序遍历的基础上稍作修改。

void flatten(TreeNode *root) 
{
	if(root==NULL) return ;

	TreeNode *x=NULL;
	stack<TreeNode*> stk;
	stk.push(root);
	while(!stk.empty())
	{
                x=stk.top();
		stk.pop();
		if(x->right !=NULL) stk.push(x->right);
		if(x->left !=NULL) stk.push(x->left);
		
		x->left=NULL;                          //增加的地方；
		if(!stk.empty()) x->right=stk.top();
		else x->right=NULL;
	}
}

扩展：按照中序遍历的顺序将二叉树转换为双向链表。

其方法也是在中序遍历的基础上修改节点。

TreeNode * bintreeToList2(TreeNode *x)
{
	stack<TreeNode *> stk;
	TreeNode *node=x;
	TreeNode *res=NULL;
	TreeNode *pre=NULL;

	int flag=1;

	while(node != NULL || !stk.empty())
	{
		while( node !=NULL)
		{
			stk.push(node);
			node=node->left;
		}
		node=stk.top();
		if( flag==1) 
		{
			res=node;		//保存头结点
			flag=0;
		}
		if(pre!=NULL)
		pre->right=node;		//就是这里前一个节点指向栈的下一个，而不是原来的
		node->left=pre;

		pre=node ;			//更新前一个节点
		node=node->right ;              //查找右半部分
		stk.pop();
	}
	return res;
}