LeetCode-109.有序链表转换二叉搜索树(相关话题:深度优先)

给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定的有序链表: [-10, -3, 0, 5, 9],

一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:

      0
     / \
   -3   9
   /   /
 -10  5

解题思路:

同leetcode-108(https://blog.csdn.net/weixin_38823568/article/details/82982853),不同之处在于,需先遍历链表获取链表长度

Java代码:

class Solution {
    public TreeNode sortedListToBST(ListNode head) {
        int len = 0;
        ListNode p = head;
        while(null != p){
            len++;
            p = p.next;
        }
        return dfs(head, 0, len);
    }
    private TreeNode dfs(ListNode head, int s, int e){
        if(s == e)
            return null;

        TreeNode root;
        if(e-1 == s){
            root = new TreeNode(head.val);
        } else {
            int p = (e-s)/2, q = (e-s)%2;
            ListNode m = head;
            for(int i = 0; i < p; i++)
                m = m.next;
            root = new TreeNode(m.val);
            root.left = dfs(head, s, s+p);
            root.right = dfs(m.next, s+p+1, e);
        }

        return root;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/82982920