108.Convert Sorted Array to Binary Search Tree
108.Convert Sorted Array to Binary Search Tree
难度:Easy
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
1
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
2
3
0
4
/ \
5
-3 9
6
/ /
7
-10 5
Copied!
递归求解,代码如下:
1
/**
2
* Definition for a binary tree node.
3
* struct TreeNode {
4
* int val;
5
* TreeNode *left;
6
* TreeNode *right;
7
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
8
* };
9
*/
10
class Solution {
11
public:
12
TreeNode* sortedArrayToBST(vector<int>& nums) {
13
if(nums.empty()) return NULL;
14
TreeNode* root = new TreeNode(0);
15
int t=nums.size();
16
if(t == 1) {
17
root->val=nums[0];
18
return root;
19
}
20
if(t==2)
21
{
22
root->val=nums[1];
23
root->left=new TreeNode(nums[0]);
24
//cout<<root->val<<endl;
25
return root;
26
}
27
if(t==3)
28
{
29
root->val=nums[1];
30
root->left=new TreeNode(nums[0]);
31
root->right=new TreeNode(nums[2]);
32
return root;
33
}
34
t=t/2;
35
vector<int>n1(nums.begin(),nums.begin()+t),n2(nums.begin()+t+1,nums.end());
36
//cout<<n1.size()<<endl;
37
// cout<<n2.size()<<endl;
38
root->val=nums[t];
39
root->left=sortedArrayToBST(n1);
40
root->right=sortedArrayToBST(n2);
41
return root;
42
}
43
};
Copied!
Copy link