二叉树构建

105. 从前序与中序遍历序列构造二叉树

    // Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]

    // Output: [3,9,20,null,null,15,7]

思路：    

（1）前序的第一个是根

（2）在中序中找到根。根据中序中的根和strat的distance，可以把前序和中序分为根、左半部[)和右半部[)三段

（3）分治递归

其他：用STL模板函数next，distance计算迭代器下一位置，及两个迭代器之间距离.CSDN

struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode() : val(0), left(nullptr), right(nullptr) {}
	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
	TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
class Solution {
public:
    // lc105
    // Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
    // Output: [3,9,20,null,null,15,7]
    // 前序的第一个是根
    // 在中序中找到根。根据中序中的根和strat的distance，可以把前序和中序分为根、左半部[)和右半部[)三段
    // 分治递归
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        TreeNode *root = nullptr;
        root = BuildTreeDfs(preorder.begin(), preorder.end(), inorder.begin(), inorder.end());
        return root;
    }
    TreeNode* BuildTreeDfs(vector<int>::iterator priItStart, vector<int>::iterator priItEnd,
        vector<int>::iterator inItStart, vector<int>::iterator inItEnd)
    {
        // 终止条件
        if (priItStart == priItEnd || inItStart == inItEnd) {
            return nullptr;
        }

        TreeNode *root = new TreeNode(*priItStart);
        vector<int>::iterator rootPostFromIn = find(inItStart, inItEnd, *priItStart);
        // 分
        // root->left = BuildTreeDfs(priItStart + 1, priItStart + 1 + distance(inItStart, rootPostFromIn),
        //                           inItStart, rootPostFromIn);
        // root->right = BuildTreeDfs(priItStart + 1 + distance(inItStart, rootPostFromIn), priItEnd,
        //                            rootPostFromIn + 1, inItEnd);
        root->left = BuildTreeDfs(next(priItStart), next(priItStart) + distance(inItStart, rootPostFromIn),
                                  inItStart, rootPostFromIn);
        root->right = BuildTreeDfs(next(priItStart) + distance(inItStart, rootPostFromIn), priItEnd,
                                   next(rootPostFromIn), inItEnd);
        // 治
        return root;
    }
};

106. 从中序与后序遍历序列构造二叉树

    // Input: 中序遍历 inorder = [9,3,15,20,7], 后序遍历 postorder = [9,15,7,20,3]

    // Output: [3,9,20,null,null,15,7]

（1）后序的最后一个是根

（2）在中序中找到根。根据中序中的根和strat的distance，可以把中序和后序分为根、左半部[)和右半部[)三段

（3）分治递归

    TreeNode* buildTreeInPost(vector<int>& inorder, vector<int>& postorder) {
        TreeNode *root = nullptr;
        root = BuildTreeDfsInPost(inorder.begin(), inorder.end(), postorder.begin(), postorder.end());
        return root;
    }
    TreeNode* BuildTreeDfsInPost(vector<int>::iterator inItStart, vector<int>::iterator inItEnd,
        vector<int>::iterator postItStart, vector<int>::iterator postItEnd)
    {
        // 终止条件
        if (inItStart == inItEnd || postItStart == postItEnd) {
            return nullptr;
        }

        TreeNode *root = new TreeNode(*prev(postItEnd));
        vector<int>::iterator rootPostFromIn = find(inItStart, inItEnd, *prev(postItEnd));
        int leftSize = distance(inItStart, rootPostFromIn);
        // 分
        root->left = BuildTreeDfsInPost(inItStart, rootPostFromIn, postItStart, next(postItStart, leftSize));
        root->right = BuildTreeDfsInPost(next(rootPostFromIn), inItEnd, next(postItStart, leftSize), prev(postItEnd));
        // 治
        return root;
    }