力扣 108. 将有序数组转换为二叉搜索树

104 阅读1分钟

🔗 leetcode.cn/problems/co…

题目

  • 给一个有序数组,转换成一颗平衡二叉搜索树
  • 平衡二叉搜索树中,所有节点满足,左孩子比节点值小,右孩子比节点值大,节点的高度差不超过 1

思路

  • 递归,二分

代码

/**
 * 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:
    TreeNode* avl(vector<int>& nums, int l, int r) {
        if (r < l) return nullptr;
        if (l > r) return nullptr;
        if (l == r) {
            TreeNode* node = new TreeNode(nums[l]);
            return node;
        }
        int mid = (l + r) / 2;
        TreeNode* node = new TreeNode(nums[mid]);
        node->left = avl(nums, l, mid - 1);
        node->right = avl(nums, mid+1, r);
        return node;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return avl(nums, 0, nums.size() - 1);
    }
};