剑指 Offer 07. 重建二叉树

最新推荐文章于 2021-03-15 15:07:58 发布

原创最新推荐文章于 2021-03-15 15:07:58 发布 · 101 阅读

0 ·

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

标签

#二叉树 #递归算法

剑指Offer总结专栏收录该内容

41 篇文章

订阅专栏

这篇博客探讨了一种利用前序遍历和中序遍历构建二叉树的高效递归方法。通过建立映射来快速查找中序遍历中的根节点位置，然后划分左右子树并递归构造，实现了在C++中24ms的执行时间和24.9MB的内存消耗。这种方法避免了重复查询，显著提高了性能。

题目链接: leetcode.

已知前序遍历与中序遍历，可以通过前序遍历得到根节点
再通过中序遍历得到左右子树的元素个数，再回到前序划分左右子树
划分得到的左右子树又知道新的前序遍历和中序遍历
所以使用递归

使用map降低查询中序遍历中根节点索引的时间复杂度

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

/*
执行用时：24 ms, 在所有 C++ 提交中击败了87.82%的用户
内存消耗：24.9 MB, 在所有 C++ 提交中击败了94.60%的用户
*/
class Solution {
public:
	unordered_map<int, int> M; 
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
    	for(int i = 0;i < inorder.size();i++)
    	{
    		M[inorder[i]] = i;
    	}
    	//递归参数：前序遍历，中序遍历，当前根节点在前序遍历中的位置，根节点的子树在 *中序遍历* 中的位置范围 
    	return build(preorder, inorder, 0, 0, inorder.size() - 1); 
    }
    TreeNode* build(vector<int>& preorder, vector<int>& inorder, int root, int minn, int maxx)
    {
    	if(minn > maxx)
    	{
    		return nullptr;
    	}
    	TreeNode* node = new TreeNode(preorder[root]);//太蠢了，这里不能再叫root
    	int root_index = M[preorder[root]];
    	//左子树的节点数 = root_index - minn 
    	//左子树的根节点在前序遍历中为 root+1，左子树的范围是 minn到 root_index-1 
    	node -> left = build(preorder, inorder, root + 1, minn, root_index - 1);
		//右子树的根节点在前序遍历中的位置为 root + 左子树的节点数 + 1 
		//右子树的根节点在前序遍历中的位置为 root + root_index - minn + 1
		node -> right = build(preorder, inorder, root + root_index - minn + 1, root_index + 1, maxx);
		return node;
    }
};