数据结构与算法学习day21-二叉树

一、从中序与后序遍历序列构造二叉树

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.确定单层递归的逻辑

1. 先要找到数组中最大的值和对应的下标， 最大的值构造根节点，下标用来下一步分割数组。
2. 最大值所在的下标左区间 构造左子树，这里要判断maxIndex > 0，因为要保证左区间至少有一个数值。
3. 最大值所在的下标右区间 构造右子树，判断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;
    }
};