leetcode654

题目:

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

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

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

Example 1:

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

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

注意:

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

解题思路:

找出最大值,以及最大值的位置,分成左右两个数组,用递归的思路解决。

代码:

class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        if(nums.empty()) return NULL;
        int mx = INT_MIN, mx_index;
        for(int i = 0;i < nums.size(); i++){
            if(nums[i] > mx){
                mx = nums[i];
                mx_index = i;
            }
        }
        vector<int> leftArr = vector<int>(nums.begin(), nums.begin()+mx_index);
        vector<int> rightArr = vector<int>(nums.begin()+mx_index+1, nums.end());
        TreeNode* node = new TreeNode(mx);
        node->left = constructMaximumBinaryTree(leftArr);
        node->right = constructMaximumBinaryTree(rightArr);
        return node;
    }
};

相关知识:

c++中的迭代器,非常好用

posted @ 2019-05-07 14:37  yxl2019  阅读(89)  评论(0编辑  收藏  举报