把排序数组转换为高度最小的二叉搜索树

来源:互联网 发布:mac os 10.12.6下载 编辑:程序博客网 时间:2024/06/15 13:05

问题描述:给一个排序数组(从小到大),将其转换为一棵高度最小的排序二叉树。

样例:

给出数组 [1,2,3,4,5,6,7], 返回

     4   /   \  2     6 / \    / \1   3  5   7
实现思路:首先找到这个数组的中值,将这个中值赋值给根节点,此时中值以左的数据都比中值小了,运用递归用这段数组构建左子树;中值以右的数据都比中值大,在运用递归构建右子树。

实现代码:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param A: A sorted (increasing order) array
     * @return: A tree node
     */
    TreeNode *ToBST(vector<int> &T,int a,int c)
    {
        if(a>c)
        return NULL;
        else{
            int b=(a+c)/2;
            TreeNode *root=new TreeNode(T[b]);
            root->left=ToBST(T,a,b-1);
            root->right=ToBST(T,b+1,c);
            return root;
        }
    }
    TreeNode *sortedArrayToBST(vector<int> &A) {
        // write your code here
        if(A.empty())
        return NULL;
        else{
            int begin=0,end=A.size()-1,mid=(begin+end)/2;
            TreeNode *root=new TreeNode(A[mid]);
            root->left=ToBST(A,begin,mid-1);
            root->right=ToBST(A,mid+1,end);
            return root;
        }
    }
};

做题感想:刚看这个题目,不知道二叉搜索树是什么概念,然后在网上搜了一下,自己确实想不出来。

0 0