Discuss

#include <iostream>
#include <algorithm>
#include <vector>


/**
* Definition for a binary tree node.
* struct TreeNode {
*     int val;
*     TreeNode *left;
*     TreeNode *right;
*     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/

using std::vector;
using std::find;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution
{
private:
    TreeNode* buildTree(vector<int>::iterator PreBegin, vector<int>::iterator PreEnd,
                        vector<int>::iterator InBegin, vector<int>::iterator InEnd)
    {
        if (PreBegin == PreEnd)
        {
            return NULL;
        }

        int HeadValue = *PreBegin;
        TreeNode *HeadNode = new TreeNode(HeadValue);

        vector<int>::iterator LeftEnd = find(InBegin, InEnd, HeadValue);
        if (LeftEnd != InEnd)
        {
            HeadNode->left = buildTree(PreBegin + 1, PreBegin + (LeftEnd - InBegin) + 1,
                             InBegin, LeftEnd);
        }

        HeadNode->right = buildTree(PreBegin + (LeftEnd - InBegin) + 1, PreEnd,
                                LeftEnd + 1, InEnd);

        return HeadNode;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)
    {
        if (preorder.empty())
        {
            return NULL;
        }

        return buildTree(preorder.begin(), preorder.end(), inorder.begin(),
                         inorder.end());

    }
};

版权声明：本文为博主原创文章，未经博主允许不得转载。