Convert Sorted List to Binary Search Tree (LeetCode)

Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.

用start和end两个point来记录当前的sublist,每次取中间值create new TreeNode后,用递归分别创建left subtree和right subtree。

值得注意的是这条语句:root.right = helper(run.next,0,end-mid-1)

之前我用的是root.right = helper(head, mid+1, end),意味着对于每次寻找sublist的中间节点,都要从头开始遍历,结果超时了,稍微改动一下即可。


另外还有个地方没想明白,第一次做这题的时候,没有用两个point,想到的是每次遍历sublist,找到中间节点,然后切断中间节点和前一个节点间的链接,例如: 1->2->3,遍历list找到中间节点为2,然后设置1.next=null,同样用递归root.left = fun(1), root.right = fun(3),可是问题是,为了切断1和2之间的链接,每次要找的实际是中间节点的前一个节点,结果各种corner case出错,比如只有1个节点怎么办,只有两个怎么办,等等。。最后索性放弃,不过这个方法应该是可行的,下次再试试。

public TreeNode sortedListToBST(ListNode head) {
        if(head==null)
            return null;
        int len = 0;
        ListNode run = head;
        while(run!=null){
            run=run.next;
            len++;
        }
        return helper(head, 0, len-1);
    }
    
    public TreeNode helper(ListNode head,int start, int end){
        if(start>end)
            return null;
        int mid = (start+end)/2, cnt = 0;
        ListNode run = head;
        while(cnt < mid){
            run = run.next;
            cnt++;
        }
        TreeNode root = new TreeNode(run.val);
        root.left = helper(head, start,mid-1);
        root.right = helper(run.next,0,end-mid-1);
        return root;
    }


【Solution】 To convert a binary search tree into a sorted circular doubly linked list, we can use the following steps: 1. Inorder traversal of the binary search tree to get the elements in sorted order. 2. Create a doubly linked list and add the elements from the inorder traversal to it. 3. Make the list circular by connecting the head and tail nodes. 4. Return the head node of the circular doubly linked list. Here's the Python code for the solution: ``` class Node: def __init__(self, val): self.val = val self.prev = None self.next = None def tree_to_doubly_list(root): if not root: return None stack = [] cur = root head = None prev = None while cur or stack: while cur: stack.append(cur) cur = cur.left cur = stack.pop() if not head: head = cur if prev: prev.right = cur cur.left = prev prev = cur cur = cur.right head.left = prev prev.right = head return head ``` To verify the accuracy of the code, we can use the following test cases: ``` # Test case 1 # Input: [4,2,5,1,3] # Output: # Binary search tree: # 4 # / \ # 2 5 # / \ # 1 3 # Doubly linked list: 1 <-> 2 <-> 3 <-> 4 <-> 5 # Doubly linked list in reverse order: 5 <-> 4 <-> 3 <-> 2 <-> 1 root = Node(4) root.left = Node(2) root.right = Node(5) root.left.left = Node(1) root.left.right = Node(3) head = tree_to_doubly_list(root) print("Binary search tree:") print_tree(root) print("Doubly linked list:") print_list(head) print("Doubly linked list in reverse order:") print_list_reverse(head) # Test case 2 # Input: [2,1,3] # Output: # Binary search tree: # 2 # / \ # 1 3 # Doubly linked list: 1 <-> 2 <-> 3 # Doubly linked list in reverse order: 3 <-> 2 <-> 1 root = Node(2) root.left = Node(1) root.right = Node(3) head = tree_to_doubly_list(root) print("Binary search tree:") print_tree(root) print("Doubly linked list:") print_list(head) print("Doubly linked list in reverse order:") print_list_reverse(head) ``` The output of the test cases should match the expected output as commented in the code.
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值