剑指 Offer 07. 重建二叉树

最新推荐文章于 2024-09-14 18:52:56 发布

nuita

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量87

点赞数

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_25155231/article/details/128758839

版权

该文介绍了如何通过给定的前序遍历和中序遍历数组来构建二叉树。方法包括两种递归策略，一种基于子树个数，另一种利用哈希映射辅助。这两种方法都利用了递归和二叉树的性质。

摘要由CSDN通过智能技术生成

//中序遍历左子树在前序遍历中个数是相同的
//比如preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
//右子树有3个 也就是最后三个
void createTree(int start, int end, TreeNode*& head, vector<int>& preorder, vector<int>& inorder, int index) {
	if (start > end)return;

	int mid = -1;
	head = new TreeNode(preorder[index]);
	for (int i = start; i <= end; i++) {
		if (inorder[i] == preorder[index]) {
			mid = i;
			break;
		}
	}
	createTree(start, mid - 1, head->left, preorder, inorder, index + 1);
	createTree(mid + 1, end, head->right, preorder, inorder, index + mid - start + 1);

}

class Solution {
public:
	TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
		TreeNode* ans = NULL;
		createTree(0, preorder.size() - 1, ans, preorder, inorder, 0);
		return ans;
	}
};

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        this->preorder = preorder;
        for(int i = 0; i < inorder.size(); i++)
            dic[inorder[i]] = i;
        return recur(0, 0, inorder.size() - 1);
    }
private:
    vector<int> preorder;
    unordered_map<int, int> dic;
    TreeNode* recur(int root, int left, int right) { 
        if(left > right) return nullptr;                        // 递归终止
        TreeNode* node = new TreeNode(preorder[root]);          // 建立根节点
        int i = dic[preorder[root]];                            // 划分根节点、左子树、右子树
        node->left = recur(root + 1, left, i - 1);              // 开启左子树递归
        node->right = recur(root + i - left + 1, i + 1, right); // 开启右子树递归
        return node;                                            // 回溯返回根节点
    }
};

//作者：Krahets
//链接：https://leetcode.cn/problems/zhong-jian-er-cha-shu-lcof/solutions/100091/mian-shi-ti-07-zhong-jian-er-cha-shu-di-gui-fa-qin/