leetcode Flatten Binary Tree to Linked List

最新推荐文章于 2019-09-19 12:26:09 发布

原创最新推荐文章于 2019-09-19 12:26:09 发布 · 676 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

c语言

当前文章被收录于：

c语言

14 篇文章 0 人学习

订阅专栏查看详情

leetcode

当前文章被收录于：

leetcode

5 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一种非递归方法实现二叉树的扁平化处理，即将二叉树转换为带有单边链表的形式，以便于进行后续的遍历操作。通过使用栈来辅助实现，确保了树的前序遍历特性。

我用的方法是非递归的前序遍历二叉树

class Solution {
public:
	void flatten(TreeNode *root) {
		if (!root || (!root->left && !root->right))
			return;
		stack s;
		if (root->right)
			s.push(root->right);
		if (root->left)
			s.push(root->left);		

		root->left = NULL;

		TreeNode *pCurrent = root, *pTemp;
		while (!s.empty()){
			pTemp = s.top();
			s.pop();

			if (pTemp->right)
				s.push(pTemp->right);
			if (pTemp->left)
				s.push(pTemp->left);			

			pTemp->left = NULL;
			pCurrent->right = pTemp;
			pCurrent = pTemp;
		}
	}	
};