1. 程式人生 > >C++詳解Leetcode:106. Construct Binary Tree from Inorder and Postorder Traversal

C++詳解Leetcode:106. Construct Binary Tree from Inorder and Postorder Traversal

原題

這裡寫圖片描述

思路

通過二叉樹的中序遍歷和後序遍歷來構建二叉樹,通過遞迴可以很簡單的解決

code

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector
<int>
& inorder, vector<int>& postorder) { int l1 = 0; int l2 = 0; int r1 = inorder.size() - 1; int r2 = postorder.size() - 1; TreeNode *s = (TreeNode*)malloc(sizeof(TreeNode)); //s = NULL; s = CreateBT(inorder, postorder, l1, r1, l2, r2); return
s; } TreeNode* CreateBT(vector<int>& inorder, vector<int>& postorder, int l1, int r1, int l2, int r2) { TreeNode* s; int i; if (l1 > r1) return NULL; s = (TreeNode*)malloc(sizeof(TreeNode)); s->left = s->right = NULL; for
(i = l1; i <= r1; i++) { if (inorder[i] == postorder[r2]) { break; } } s->val = inorder[i]; s->left = CreateBT(inorder, postorder, l1, i - 1, l2, l2 + i -l1 - 1); s->right = CreateBT(inorder, postorder, i + 1, r1, l2 + i - l1, r2 - 1); return s; } };