题解 | #将升序数组转化为平衡二叉搜索树#

将升序数组转化为平衡二叉搜索树

http://www.nowcoder.com/practice/7e5b00f94b254da599a9472fe5ab283d

将升序数组转化为平衡二叉搜索树
题解一:二分+递归
题解思路: 利用BST中序遍历为升序序列的特性。
BST中序遍历:
图片说明
思路分析: 为了保持平衡,每次选择区间的中间值作为根节点。
递归分析:
递归边界: (left > right) 返回NULL;
递归过程: 选取中间值作为根节点,root = mid = (left+right+1)/2; 递归构建左右子树。

复杂度分析:
时间复杂度:O(N) 节点数N
空间复杂度:O(logN)

实现如下:

class Solution {
public:
    /**
     *
     * @param num int整型vector
     * @return TreeNode类
     */
    TreeNode* creat_BST(vector<int>& num,int left,int right){
        if(left>right) return NULL;  // 递归边界
        int mid = (left+right+1)/2;   //选取中间位置的值作为根节点
        TreeNode* root = new TreeNode(num[mid]);  //创建树的根节点
        root -> left  = creat_BST(num, left, mid-1);  // 构建左子树
        root-> right = creat_BST(num, mid+1, right);  // 构建右子树
        return root;
    }
    TreeNode* sortedArrayToBST(vector<int>& num) {
        // write code here
        if(num.size() ==0 ) return NULL; // 如果数组为空,返回空树
        return creat_BST(num, 0, num.size()-1); //  递归建树
    }
};

题解二: 迭代
题解思路: 与题解一相同,不过将递归转为迭代的写法.依然是分割的思想。
分析: 使用一个栈来保存左右子树边界,一个栈用来保存遍历的节点。
复杂度度分析:
时间复杂度:O(N),遍历整个节点数N
空间复杂度:O(logN)
实现如下:

class Solution {
public:
    TreeNode* sortedArrayToBST(vector<int>& num) {
        if(num.size() ==0 ) return NULL;
        stack<TreeNode*> node_st; // 用来保存构造的节点
        stack<int> lr_index; // 用来保存边界
        TreeNode* root = new TreeNode(0); //根节点入栈
        node_st.push(root);
        lr_index.push(0);  // 将左边界入栈
        lr_index.push(num.size()-1); // 右边界入栈
        while(node_st.size()){
            int right = lr_index.top();lr_index.pop();  //取出左边界
            int left = lr_index.top();lr_index.pop();   //取出右边界
            int mid = (right+left+1)/2; //取中间值作为根节点的值
            TreeNode* tmp = node_st.top();   //取出当前节点
            node_st.pop();
            tmp->val = num[mid];   //将中间值赋予当前节点的值

            //划分左子树
            if(left<= mid-1){
                tmp->left = new TreeNode(0);
                node_st.push(tmp->left);
                lr_index.push(left);
                lr_index.push(mid-1);
            }
            //划分右子树
            if(right>= mid+1){
                tmp->right = new TreeNode(0);
                node_st.push(tmp->right);
                lr_index.push(mid+1); //
                lr_index.push(right);
            }
        }
        return root;
    }
};
牛客网编程题题解 文章被收录于专栏

本专栏记录在牛客网上AC的每一题,写下题解。 未来2年完成2000编程题的题解。 2021.12.29更新,最进准备毕设,断更了,会尽快做完毕设,继续做这一件事情

全部评论

相关推荐

头像
11-09 12:17
清华大学 C++
out11Man:小丑罢了,不用理会
点赞 评论 收藏
分享
9 4 评论
分享
牛客网
牛客企业服务