题目 有序数组转化为二叉搜索树
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
题解 找中点递归
题目给的数组已经是升序得,找中间得位置当根节点,左边的为左子树,右边的为右子树,再去找左边的数组得中点,右边数组的中点,依次递归。
/**
* 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:
TreeNode* BuildTree(vector<int>& nums, int left, int right){
if(left > right)
return nullptr;
int mid = (left + right) / 2;
TreeNode* root = new TreeNode(nums[mid]);
root->left = BuildTree(nums, left, mid - 1);
root->right = BuildTree(nums, mid + 1, right);
return root;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
int len = nums.size();
return BuildTree(nums, 0, len - 1);
}
};
题目 有序链表转化为高度平衡的二叉树
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
题解 找中点递归
难点:链表不像数组那样可以直接去到中点,因此可以利用快慢指针,一个指针一次走一步,一个指针一次走两步,当快指针走到终点的时候,慢指针就在中间。
注意递归结束的条件
if(head == end)
return nullptr;
完整代码:
/**
* 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:
TreeNode* BuildTree(ListNode* head, ListNode* end){
if(head == end)
return nullptr;
ListNode* slow = head;
ListNode* fast = head;
while(fast != end && fast->next != end){
fast = fast->next;
slow = slow->next;
fast = fast->next;
}
TreeNode *root = new TreeNode(slow->val);
root->left = BuildTree(head, slow);
root->right = BuildTree(slow->next, end);
return root;
}
TreeNode* sortedListToBST(ListNode* head) {
if(!head)
return nullptr;
ListNode* end = head;
while(end!= nullptr){
end = end->next;
}
return BuildTree(head, end);
}
};