LeetCode 105. 已知前序&中序求二叉树

最新推荐文章于 2022-10-23 21:19:17 发布

原创最新推荐文章于 2022-10-23 21:19:17 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

1849 篇文章

订阅专栏

本文深入探讨了二叉树的重构算法，通过前序和中序遍历序列重建二叉树结构。提供了两种实现方法：递归法和迭代法，并详细解释了每种方法的实现细节和工作原理。

文章目录

- 1. 题目
- 2. 解题

1. 题目

在这里插入图片描述
类似POJ 2255 Tree Recovery

2. 解题

在这里插入图片描述

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        unordered_map<int,int> m;//哈希表
        for(int i = 0; i < inorder.size(); ++i) 
        {
        	m[inorder[i]] = i;//方便后面查找位置
        }
        return build(preorder, inorder,0,preorder.size()-1,0,inorder.size()-1,m);
    }
    TreeNode* build(vector<int>& preorder, vector<int>& inorder, int pS, int pE, int iS, int iE, unordered_map<int,int> &m)
    {
    	if(pS > pE)
    		return NULL;
    	TreeNode *root = new TreeNode(preorder[pS]);
    	int leftlen = m[preorder[pS]]-iS;
    	root->left = build(preorder,inorder,pS+1,pS+leftlen,iS,m[preorder[pS]]-1,m);
    	root->right = build(preorder,inorder,pS+1+leftlen,pE,m[preorder[pS]]+1,iE,m);
    	return root;
    }
};

别人的循环解法，很牛，也很难理解

class Solution {//别人写的，循环法
public:
    TreeNode* buildTree(vector<int>& pre, vector<int>& in) 
    {
		if (pre.empty()) 
			return NULL;
		stack<TreeNode*> S;
		TreeNode* root = new TreeNode(pre[0]);
		S.push(root);
		for (int i = 1, j = 0; i < pre.size(); i++) 
		{  // i-前序序号，j-中序序号
			TreeNode *back = NULL, *cur = new TreeNode(pre[i]);
			while (!S.empty() && S.top()->val == in[j]) 
			{
				back = S.top(), 
				S.pop(), 
				j++;
			}	
			if (back)
			  	back->right = cur;
			else
			  	S.top()->left = cur;
			S.push(cur);
		}
		return root;
    }
};