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

来源:互联网 发布:重复犯错歌词意思 知乎 编辑:程序博客网 时间:2024/06/06 03:13

题目:

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

 注意事项

There may exist multiple valid solutions, return any of them.

样例

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

     4   /   \  2     6 / \    / \1   3  5   7

思路:

将给出的向量从中间一分为二,小的一部分位于当前根节点的左子树位置,大的一部分位于当前根节点的右子树位置。运用递归直至向量中没有可以分的元素为止。在要实现的函数里令 int begin=0;
        int end=A.size()-1;建立初始值,便于调用之前定义的函数求解问题。

代码:

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

感想:

这个题不是我自己想出来的,一开始理解错了题意,完全按照层次遍历的思路去放节点。后来搜了一下别人的代码,理解了一下自己按照理解的打上去了,感觉这个想法很奇妙。十分值得学习。

0 0
原创粉丝点击