leetcode-Construct Binary Tree from Inorder and Postorder Traversal

Given inorder and postorder traversal of a tree, construct the binary tree.
Note: You may assume that duplicates do not exist in the tree.
이 문제는 귀속으로 풀기가 어렵지 않다. 두 갈래 나무의 서열과 후속으로 훑어보는 특징에 따라 풀면 된다. 비귀속으로 어떻게 풀어야 할지 모르겠다.
/**
 * Definition for binary tree
 * 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) {
        if (inorder.empty())
            return NULL;
        return buildTree(inorder, 0, inorder.size()-1, postorder, 0, postorder.size()-1);
    }
private:
    TreeNode *buildTree(vector<int> &inorder, int istart, int iend, 
                     vector<int> &postorder, int pstart, int pend) {
        if (istart == iend)
            return new TreeNode(inorder[istart]);
        
        int i;
        for (i=istart; i<=iend; ++i) {
            if (inorder[i]==postorder[pend]) {
                break;
            }
        }
        
        TreeNode *root = new TreeNode(inorder[i]);
        
        if (i>istart)
            root->left = buildTree(inorder, istart, i-1, postorder, pstart, pstart+i-istart-1);
        
        if (i<iend)
            root->right = buildTree(inorder, i+1, iend, postorder, pstart+i-istart, pend-1);
        
        return root;
    }
    
};

좋은 웹페이지 즐겨찾기