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;
}
};