109. 有序链表转换二叉搜索树 - 力扣(LeetCode)
LeetCode第 108 题:将有序数组转化为二叉搜索树(C++)_zj-CSDN博客的进阶。
一个高度平衡二叉树是指一个二叉树每个节点的左右两个子树的高度差的绝对值不超过 1。
108题的时候,我们可以很轻松地通过下标计算出中间节点。
可是这一题不是数组,而是链表,获取中间节点的方式并不直接,当然可以先对链表进行遍历放入数组,就转化为108题了。
class Solution {
public:
vector<int> nums;
TreeNode* create(int low, int high){
if(low > high) return NULL;
int mid = (low + high)/2;
auto root = new TreeNode(nums[mid]);
root->left = create(low, mid-1);
root->right = create(mid+1, high);
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
while(head){
nums.push_back(head->val);
head = head->next;
}
return create(0, nums.size()-1);
}
};
如果不使用外部数组,直接在链表上寻找中间节点的话,可以使用快慢指针法。由于可以生成的树有很多种,我们只需要生成一颗符合要求的就行:
快慢指针找到中点,但是依然需要每次都进行快慢指针找中点的步骤,感觉还不如使用数组。因为需要一直递归。。。
关于翻转可以看:LeetCode第 234 题:回文链表(C++)_zj-CSDN博客
这儿的思路其实就是找中点,切断然后分治。
class Solution {
public:
TreeNode* sortedListToBST(ListNode* head) {
if(!head) return NULL;
if(!head->next) return new TreeNode(head->val);
ListNode *slow = head, *fast = head;
ListNode *pre = NULL;
while(fast && fast->next){
pre = slow;
slow = slow->next;
fast = fast->next->next;
}//slow指向中点或者右中点
pre->next = NULL;//切断
TreeNode* root = new TreeNode(slow->val);
root->left = sortedListToBST(head);
root->right = sortedListToBST(slow->next);
return root;
}
};