力扣题解-654. 最大二叉树(分治法思想,递归的方式求解)

题目:654. 最大二叉树

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

二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。

示例 :

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

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1
 

提示:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

  1. divide
    找出原始数组的最大值及其所在位置pos,将原始数组按照pos位置分为左边部分和右边部分。

  2. conquer
    将pos位置的数组元素作为根节点root,然后递归地处理左边部分和右边部分,分别作为左子树left和右子树right。

  3. combine
    将二叉树拼接起来,即root->left = left,root->right = right。
    return root

代码

/**
 * 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) {
        return dfs(nums, 0, nums.size()-1);
    }

    TreeNode* dfs(vector<int>& nums, int start, int end) {
        if (start > end) {
            return NULL;
        }
        int pos = maxPos(nums, start, end);
        TreeNode* node = new TreeNode(nums[pos]);

        TreeNode* left = dfs(nums, start, pos-1);
        TreeNode* right = dfs(nums, pos+1, end);
        node->left = left;
        node->right = right;
        return node;
    }

    int maxPos(vector<int>& nums, int start, int end) {
        int max = nums[start];
        int pos = start;
        for (int i = start; i <= end; i++) {
            if (max < nums[i]) {
                max = nums[i];
                pos = i;
            }
        }
        return pos;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值