一、从中序与后序遍历序列构造二叉树
1.题目
106. 从中序与后序遍历序列构造二叉树 - 力扣(LeetCode)
2.思路
以后序(左右中)数组的最后一个元素为切割点(根节点),切割的节点切中序(左中右)数组,左边为左子树,右边为右子树。
怎么进行下一步切割?
需要找出左右子树的后序数组。
中序数组和后序数组的左右子树长度一定是相等的。根据中序切割出来的左右子树的数据长度,
再切后序数组,切出后序数组的左右子树。
一层一层切下去,每次后序数组最后一个元素就是节点元素。
class Solution {
private:TreeNode* treaversal(vector<int>& inorder, vector<int>& postorder){//如果是空的,直接返回空指针if(inorder.size() == 0) return NULL;//后续遍历数组最后以一个节点就是根节点int getroot = postorder[postorder.size()-1];TreeNode* root = new TreeNode(getroot);//终止条件if (postorder.size() == 1) return root;//开始分割中序数组,分成左右两个数组//左闭右开区间int i;for(i = 0;i < inorder.size(); i++){if(inorder[i] == getroot) break;}vector<int> left_inorder(inorder.begin(),inorder.begin()+i);vector<int> right_inorder(inorder.begin()+i+1,inorder.end());//舍弃后序数组最后一个分割元素postorder.resize(postorder.size()-1);//根据分割完的左右数组大小分割后序遍历的数组vector<int> left_postorder(postorder.begin(),postorder.begin()+left_inorder.size());vector<int> right_postorder(postorder.begin()+left_inorder.size(),postorder.end());root->left = treaversal(left_inorder,left_postorder);root->right = treaversal(right_inorder,right_postorder);return root;}public:TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {return treaversal(inorder,postorder);}
};
二、最大二叉树
1.题目
654. 最大二叉树 - 力扣(LeetCode)
2.思路
构造树一般采用的是前序遍历,因为先构造中间节点,然后递归构造左子树和右子树。
1.确定递归函数的参数和返回值
2.确定终止条件
题目中说了输入的数组大小一定是大于等于1的,所以我们不用考虑小于1的情况,那么当递归遍历的时候,如果传入的数组大小为1,说明遍历到了叶子节点了。那么应该定义一个新的节点,并把这个数组的数值赋给新的节点,然后返回这个节点。 这表示一个数组大小是1的时候,构造了一个新的节点,并返回。
3.确定单层递归的逻辑
- 先要找到数组中最大的值和对应的下标, 最大的值构造根节点,下标用来下一步分割数组。
- 最大值所在的下标左区间 构造左子树,这里要判断maxIndex > 0,因为要保证左区间至少有一个数值。
- 最大值所在的下标右区间 构造右子树,判断maxIndex < (nums.size() - 1),确保右区间至少有一个数值。
总体代码:
class Solution {
public:TreeNode* constructMaximumBinaryTree(vector<int>& nums) {//结束条件TreeNode* node = new TreeNode(0);if(nums.size() == 1){node->val = nums[0];return node;}//int max = 0;int maxIndex = 0;for(int i = 0; i < nums.size();i++){if(nums[i] > max){max = nums[i];maxIndex = i;}}//根节点node->val = max;//左if(maxIndex > 0){vector<int> leftnums(nums.begin(),nums.begin()+maxIndex);node->left = constructMaximumBinaryTree(leftnums);}//右if(maxIndex < nums.size()-1){vector<int> rightnums(nums.begin()+maxIndex+1,nums.end());node->right = constructMaximumBinaryTree(rightnums); }return node;}
};