leetcode_力扣_654. 最大二叉树

题目描述

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:

1. 创建一个根节点，其值为 nums 中的最大值。
2. 递归地在最大值 左边 的 子数组前缀上 构建左子树。
3. 递归地在最大值 右边 的 子数组后缀上 构建右子树。

返回 nums 构建的 最大二叉树 。

示例

• 示例 1：
  
  
• 示例 2：
  
  

提示

• 1 <= nums.length <= 1000
• 0 <= nums[i] <= 1000
• nums 中的所有整数 互不相同

解题思路

1. 递归处理，注意细节；
2. 代码里面有注释。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int find_max_idx(vector<int> nums, int left, int right){	// 寻找数组给定范围中最大元素的下标
        int ret = left;

        int i = 0;
        for(i = left; i <= right; i++){
            if(nums[i] > nums[ret]){
                ret = i;
            }
        }

        return ret;
    }
    
    int myfunc(TreeNode **root, vector<int> nums, int left, int right){		
		// 递归构造函数，注意传入的 **root，记得在堆上开辟节点，并做好赋值
		
        if(left > right){	// 递归终止条件
            return 0;
        }

        int max_idx = find_max_idx(nums, left, right);
		
		// 处理当前节点
        TreeNode *temp = new TreeNode;
        temp->val = nums[max_idx];
        temp->left = NULL;
        temp->right = NULL;
		
		// 递归处理左右节点
        myfunc(&(temp->left), nums, left, max_idx - 1);
        myfunc(&(temp->right), nums, max_idx + 1, right);

		// 做好二级指针的赋值
        *root = temp;

        return 0;
    }

    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {

        int nums_len = nums.size();

        TreeNode *ret = NULL;

        myfunc(&ret, nums, 0, nums_len - 1);	// 为了统一处理，自己写个递归，此处传入的是根节点的二级指针

        return ret;
    }
};

结果截图

总结

• 二叉树的基本操作，感觉和二叉树相关的题大多数都用到了递归；
• 占用了好多内存，运行时间也不太行，下次注意。