面试题 04.02. 最小高度树

题目描述

给定一个有序整数数组，元素各不相同且按升序排列，编写一个算法，创建一棵高度最小的二叉搜索树。

示例:
给定有序数组: [-10,-3,0,5,9],

一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：

      0 
     / \ 
   -3   9 
   /   / 
 -10  5 

思路分析

方法一：深度优先搜索+递归

二叉搜索树的中序遍历是一个递增的有序序列。如果要保证一个递增的有序序列对应二叉搜索树的深度最小，那么需要将中间结点作为根结点，然后左右区间的有序序列依次递归建立左右子树即可。

代码实现

/**
 * 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* dfs(vector<int>& nums, int l, int r){
        if(l > r)
            return NULL;
        else{
            int middle = l + (r-l) / 2;
            TreeNode *root = new TreeNode(nums[middle]);
            root->left = dfs(nums, l, middle-1);
            root->right = dfs(nums, middle+1, r);
            return root;
        }
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return dfs(nums, 0, nums.size()-1);
    }
};

复杂度