[数据结构与算法]leetcode109有序链表转换为二叉树

最新推荐文章于 2022-07-20 20:00:01 发布

肖特补失基

最新推荐文章于 2022-07-20 20:00:01 发布

阅读量137

点赞数

分类专栏：算法 leetcode 二叉树文章标签：链表二叉树算法数据结构

本文链接：https://blog.csdn.net/weixin_43515720/article/details/107623731

版权

算法 leetcode 二叉树专栏收录该内容

29 篇文章 0 订阅

订阅专栏

给定一个单链表，其中的元素按升序排序，将其转换为高度平衡的二叉搜索树。本题中，
一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表： [-10, -3, 0, 5, 9],
一个可能的答案是：[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树：
      0
     / \
   -3   9
   /   /
 -10  5
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree

class Solution {
    public ListNode findMid(ListNode head){
        ListNode prevPtr = null;
        ListNode slowPtr = head;
        ListNode fastPtr = head;
        while(fastPtr != null && fastPtr.next != null){
            prevPtr= slowPtr;
            slowPtr = slowPtr.next;
            fastPtr = fastPtr.next.next;
        }
        if(prevPtr != null) prevPtr.next = null;
        return slowPtr;
    }
    public TreeNode sortedListToBST(ListNode head) {
        if(head == null) return null;
        ListNode mid = this.findMid(head);
        TreeNode node = new TreeNode(mid.val);
        if(head == mid) return node;
        node.left = this.sortedListToBST(head);
        node.right = this.sortedListToBST(mid.next);
        return node;
    }
}

class Solution {
    private List<Integer> values;
    public Solution(){
        this.values = new ArrayList<>();
    }
    private void mapListToValue(ListNode head){
        while(head != null){
            this.values.add(head.val);
            head = head.next;
        }
    }
    public TreeNode findMid(int left,int right){
        if(left > right) return null;
        int mid = (left+right)/2;
        TreeNode node = new TreeNode(this.values.get(mid));
        if(left == right) return node;
        node.left = findMid(left,mid-1);
        node.right = findMid(mid+1,right);
        return node;
    }
    public TreeNode sortedListToBST(ListNode head){
        this.mapListToValue(head);
        return findMid(0,this.values.size()-1);
    }
}