(leetcode)654. 最大二叉树

给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下：

1. 二叉树的根是数组中的最大元素。
2. 左子树是通过数组中最大值左边部分构造出的最大二叉树。
3. 右子树是通过数组中最大值右边部分构造出的最大二叉树。

4. 通过给定的数组构建最大二叉树，并且输出这个树的根节点。

5. Example 1:

   输入: [3,2,1,6,0,5]
   输入: 返回下面这棵树的根节点：
   
         6
       /   \
      3     5
       \    / 
        2  0   
          \
           1
   

   注意:给定的数组的大小在 [1, 1000] 之间。

算法思想：递归实现，先找到最大元素建根节点，然后分别根据左右元素建左子树和右子树

/**
 * 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* constructMaximumBinaryTree(vector<int>& nums) {
        int nodeNum = nums.size();
        TreeNode* root = NULL;
        findMax(nums, 0, nodeNum, root);
        return root;
    }
    
    void findMax(vector<int>& nums, int start, int end, TreeNode* &root){
        if(end <= start)  return;
        int max = -999999;
        int max_index = -1;
        vector<int>::iterator iter_start = nums.begin();
        vector<int>::iterator iter;
        for(int i = start; i < end; i++){
            iter = iter_start + i;
            if(*iter > max){
                max = *iter;
                max_index = i;
            }
        }
        root = new TreeNode(max);//新建对象
        findMax(nums, start, max_index, root->left);
        findMax(nums, max_index+1, end, root->right);
    }
};

欢迎各位评论！！！