二叉树由有序链表和有序数组转换

最新推荐文章于 2022-02-16 12:24:16 发布

转载最新推荐文章于 2022-02-16 12:24:16 发布 · 546 阅读

leetcode 专栏收录该内容

175 篇文章

订阅专栏

本文介绍了一种算法，该算法将已排序的单链表转换为平衡的二叉搜索树。通过使用快慢指针技巧找到链表的中点，并以此作为根节点，然后递归地构造左子树和右子树。此外，还提供了一种从数组构建平衡二叉搜索树的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *sortedListToBST(ListNode *head) {
        TreeNode *root=NULL;
        if(head==NULL){
            return root;
        }else if(head->next==NULL){
            return new TreeNode(head->val);
        }
        //快慢指针找到对应的中间节点
        ListNode *pre=NULL;
        ListNode *mid=head;
        ListNode *fast=head;
        while(fast!=NULL&&fast->next!=NULL){
            pre=mid;
            mid=mid->next;
            fast=fast->next->next;
        }
        root=new TreeNode(mid->val);
        pre->next=NULL;//将中间节点以前的断开
        root->left=sortedListToBST(head); //递归左侧
        root->right=sortedListToBST(mid->next);//递归右侧
        
        return root;
    
    }
    
    TreeNode *arraytoTree(vector<int>&ivec,int start,int end){
        int mid=start+(end-start)/2;
        TreeNode *root=new TreeNode(ivec[mid]);
        if(start==end){
            return root;   
        }
        if(start<=mid-1){
            root->left=arraytoTree(ivec,start,mid-1);
        }
        if(mid+1<=end){
            root->right=arraytoTree(ivec,mid+1,end);
        }
        
        return root;
    }
    
    void addnode(TreeNode * &root,int value){
        if(root==NULL){
            root=new TreeNode(value);
        }else if(root->val>value){
            addnode(root->left,value);
        }else if(root->val<value){
            addnode(root->right,value);
        }
    }
};