给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
思路:由于是有序的,所以需要找到中间的节点作为树的根节点。并利用分治的思想,递归的调用这个函数,找到左子树和右子树的根节点。
代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
// 获取中间节点的函数
ListNode* midNode(ListNode* left, ListNode* right){
ListNode* fast = left; //定义一个跑的快的节点
ListNode* slow = left; // 定义一个跑的慢的节点
while(fast != right && fast->next != right){
fast = fast->next;
fast = fast->next; // 每次跑两步
slow = slow->next; // 每次跑一步
}
return slow;
}
TreeNode* buildTree(ListNode* left, ListNode* right){
if(left == right)
return nullptr;
// 获取中间的节点作为二叉搜索树的根节点
ListNode* mid = midNode(left, right);
TreeNode* root = new TreeNode(mid->val);
root->left = buildTree(left, mid);
root->right = buildTree(mid->next, right);
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
// 拦腰截断(高度平衡)
return buildTree(head, nullptr);
}
};
对于递归的思想还是不太熟悉,只能先看着答案先做着,应该做多点就会有感觉了吧。