Convert Sorted Array to Binary Search Tree-优快云博客

一、问题描述

Given an array where elements are sorted in ascending order, convert it to a height balanced BST.

二、思路

题意是将有序数组转换为二叉搜索树。思路很简单，递归建立二叉搜索树，将中间值作为根节点即可，然后分别递归建立左子树和右子树。在建立节点时，如果节点为空，必须申请空间，同时赋值三个域即可。还有就是在传递参数时，由于我们平时在定义二叉树时除了定义一个TreeNode节点，还会定义一个二叉树的指针，这样在传递参数时就不会感觉那么奇怪。比如本例中由于没有定义，所以传递参数时出现下面这种 *&符号连写，让自己以为自己写错了。

TreeNode* &t

三、代码

/**
 * 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:
    void createTree(TreeNode* &t,vector<int>& nums,int start,int end){
        if(start > end)
            return;
        int mid = (start + end) / 2;
        if(t == NULL){
            t = (TreeNode *)malloc(sizeof(TreeNode));
            t -> val = nums[mid];
            t -> left = NULL;
            t -> right = NULL;
        }
        createTree(t -> left, nums, start, mid - 1);
        createTree(t -> right, nums, mid + 1, end);
    }
    TreeNode* sortedArrayToBST(vector<int>& nums){
        TreeNode* root = NULL;
        int n = nums.size();
        int start = 0,end = n - 1;
        createTree(root, nums, start, end);
        return root;
    }
};