【LeetCode】Flatten Binary Tree to Linked List

最新推荐文章于 2022-09-06 16:04:14 发布

原创最新推荐文章于 2022-09-06 16:04:14 发布 · 321 阅读

0 ·

CC 4.0 BY-SA版权

算法学习同时被 2 个专栏收录

104 篇文章

订阅专栏

【LeetCode】

100 篇文章

订阅专栏

本文介绍了一种将二叉树结构通过迭代方法展平为链表的方法。具体思路是从左子树开始遍历，若存在右子树则将其压入栈中，随后将左子树指针连接到当前节点的右子树。当左子树遍历完成，从栈中弹出节点继续处理，确保每个节点的右子树指向下一个前序遍历的节点。

题目描述：

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

For example,
Given

The flattened tree should look like:

Hints:

If you notice carefully in the flattened tree, each node's right child points to the next node of a pre-order traversal.

这题目思路跟之前的用迭代法对二叉树进行后序遍历有点像。从左边开始遍历，若存在右孩子，则将右孩子压入堆栈，然后将左孩子赋值到右指针。左边遍历结束开始从栈中取出节点。代码如下：

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
	void flatten(TreeNode *root)
	{
		if (!root)
			return;
		stack<TreeNode*> store;
		TreeNode *node = root;
		while (!store.empty() || node->left || node->right)
		{
			if (node->right)
			{
				store.push(node->right);
				node->right = NULL;
			}
			if (node->left)
			{
				node->right = node->left;
				node->left = NULL;
				node = node->right;
			}
			else
			{
				node->right = store.top();
				node = store.top();
				store.pop();
			}
		}
	}
};