剑指 Offer 07. 重建二叉树
作者:互联网
剑指 Offer 07. 重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1] Output: [-1]
分治不太熟悉,很懵。。做了一半,做不下去了,代码写的很乱,之后看了解析
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; // 回溯返回根节点 } };
标签:preorder,07,Offer,int,root,二叉树,inorder,left 来源: https://www.cnblogs.com/4jiu/p/15964520.html