[geeks]Construct BST from given preorder traversal——iterative solution

最新推荐文章于 2024-08-29 09:19:11 发布

Chen_chish

最新推荐文章于 2024-08-29 09:19:11 发布

阅读量529

点赞数

CC 4.0 BY-SA版权

分类专栏： Tree

本文链接：https://blog.youkuaiyun.com/u010970771/article/details/9168435

Tree 专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种利用前序遍历数组构建二叉树的非递归方法，通过迭代过程，有效地实现了二叉树结构的重建。具体步骤包括初始化空栈，将前序数组的第一个元素作为根节点加入栈中，随后根据数组元素与栈顶元素值的关系决定左子树或右子树的连接。

步骤：

1、建立一个空栈

2、使preorder数组中的第一个元素为root，并把root推入栈中

3、重复下面步骤直到遍历完preorder数组

如果当前preorder数组中的元素大于栈中节点的值，pop栈顶元素，直到栈为空，或者栈顶节点的值大于preorder数组中的元素。令最后被pop出来的节点的右子树为当前元素，并将其push入栈。

如果当前元素小于栈顶节点的值，那么令栈顶节点的左子树为当前节点，并将其push入栈。

struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
	TreeNode *buildTree(vector<int> &preorder) {
		/*O(n)非递归*/
		stack<TreeNode*> myStack;
		TreeNode *root = new TreeNode(preorder[0]);
		myStack.push(root);
		for(int i = 1; i < preorder.size(); ++i)
		{
			TreeNode *tmp = NULL;
			while(!myStack.empty() && preorder[i] > myStack.top()->val)
				tmp = myStack.top(), myStack.pop();
			if(tmp != NULL)
			{
				tmp->right = new TreeNode(preorder[i]);
				myStack.push(tmp->right);
			}
			else
			{
				myStack.top()->left = new TreeNode(preorder[i]);
				myStack.push(myStack.top()->left);
			}
		}
		return root;
	}
};