给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9], 一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树: 0 / \ -3 9 / / -10 5
基本思路:
- 首先看到题目就能想到递归。
- 递归的子问题: 找到一个数组的中间的位置,其中间往左的数组为其左边的节点,中间往右的数组为其右边的节点。
- 再从中间往左的数组中找出找出中间位置,这个中间位置的左边为左边的节点,右边为右边的节点,以此类推。
很简单的题
class Solution {
public:
TreeNode * sortedListToBST(ListNode* head) {
vector<int> Temp;
if (head == NULL)
return NULL;
while (head != NULL) {
Temp.push_back(head->val);
head = head->next;
}
TreeNode* Res = new TreeNode(Temp[Temp.size() / 2]);
_sortedListToBST(Temp, Res, 0 , Temp.size()/2-1, Temp.size() / 2 + 1, Temp.size()-1);
return Res;
}
void _sortedListToBST(vector<int> &Temp, TreeNode* Res, int L_S, int L_E, int R_S, int R_E) {
if (L_S <= L_E) {
int L_Mid = (L_E - L_S) / 2 + L_S;
TreeNode* L_node = new TreeNode(Temp[L_Mid]);
Res->left = L_node;
_sortedListToBST(Temp, Res->left, L_S, L_Mid - 1, L_Mid + 1, L_E);
}
if (R_S <= R_E) {
int R_Mid = (R_E - R_S) / 2 + R_S;
TreeNode* R_node = new TreeNode(Temp[R_Mid]);
Res->right = R_node;
_sortedListToBST(Temp, Res->right, R_S, R_Mid - 1, R_Mid + 1, R_E);
}
return;
}
};