LeetCode 108: 将有序数组转换为二叉搜索树

LeetCode 108: 将有序数组转换为二叉搜索树

题目描述

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

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

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

解题

    要想将有序数组转换成的二叉搜索树尽可能平衡，即要维持每个节点的左右子树高度差不超过1，只需要保证每个子树的“根节点”恰好位于该子树对应的有序子序列的中点。那么，有序数组的中点元素为平衡二叉搜索树的根节点，其左边元素为根节点的左子树，右边元素为根节点的左子树，对于左右子序列而言，其中点元素为子树的“根节点”，以此“分治”。
    可以建立一个新的迭代函数，该函数提供原始数组序列，以及起始位置和终止位置，起始位置和终止位置确定了需要转化的子序列，在当前函数体中，子序列中间位置的元素为当前节点，其左边序列的迭代返回结果为其左子节点，右边序列的迭代返回结果为其右子节点。

/**
 * 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* sortedArrayToBST(vector<int>& nums) {
        if (nums.size()==0) return NULL;
        return sortedArrayToBST(nums, 0, nums.size()-1);
    }

private:
     TreeNode* sortedArrayToBST(vector<int>& nums, const int lp, const int rp) {
        if (lp > rp) return NULL;
        else{
            int curp = (lp+rp)>>1;
            TreeNode *cur_node = new TreeNode(nums[curp]);
            cur_node->left = sortedArrayToBST(nums, lp, curp-1);
            cur_node->right = sortedArrayToBST(nums, curp+1, rp);
            return cur_node;
        }
    }
};

    当然也可以不构造新的迭代函数，直接在接口函数上实现，只需要迭代过程中，构造出左子序列和右子序列，并作为参数继续迭代。

/**
 * 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* sortedArrayToBST(vector<int>& nums) {
        if (nums.size()==0) return NULL;
        if (nums.size()==1) return new TreeNode(nums[0]);
        size_t mid = nums.size() >> 1;
        TreeNode *cur_node = new TreeNode(nums[mid]);
        vector<int> left_vec(nums.begin(), nums.begin() + mid);
        vector<int> right_vec(nums.begin() + mid+1, nums.end());
        cur_node->left = sortedArrayToBST(left_vec);
        cur_node->right = sortedArrayToBST(right_vec);
        return cur_node;
    }
};