码迷,mamicode.com
首页 > 其他好文 > 详细

109. Convert Sorted List to Binary Search Tree

时间:2018-08-17 00:42:41      阅读:166      评论:0      收藏:0      [点我收藏+]

标签:while   tail   nod   dmi   exception   lin   span   ase   list   

109. Convert Sorted List to Binary Search Tree
(AC)

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
  
class Solution {
    public TreeNode sortedListToBST(ListNode head) {
      if(head == null) return null;
      return helper(head, null);
    }
  
    public TreeNode helper(ListNode head, ListNode tail){
      if(head == tail) return null;/////
    
      ListNode slow = head;
      ListNode fast = head;
      
      
      while(fast != tail && fast.next != tail){
        slow = slow.next;
        fast = fast.next.next;
      }
      
      TreeNode root = new TreeNode(slow.val);
      root.left = helper(head, slow);
      root.right = helper(slow.next, tail);
      return root;
    }
}










/// NullPointerException


class Solution {
    public TreeNode sortedListToBST(ListNode head) {
      // base cases 
      // size = 0;
      if(head == null) return null;
      // size = 1
      if(head.next == null){
        TreeNode root = new TreeNode(head.next.val);
        return root;
      }
      
      // size = 2
      if(head.next.next == null){
        TreeNode next = new TreeNode(head.next.val);
        TreeNode root = new TreeNode(head.val);
        
        root.right = next;
        return root;
      }
      
      // size = 3 
      if(head.next.next.next == null){
        TreeNode root = new TreeNode(head.next.val);
        TreeNode right = new TreeNode(head.next.next.val);
        TreeNode left = new TreeNode(head.val);
        
        root.left = left;
        root.right = right;
        
        return root;
      }
      
      // induction rules
      
      ListNode mid = findMid(head);
      ListNode rootNode = mid.next;
      ListNode nextNode = rootNode.next;
      
      mid.next = null;
      rootNode.next = null;
      
      TreeNode root = new TreeNode(rootNode.val);
      root.left = sortedListToBST(head);
      root.right = sortedListToBST(nextNode);
      
      return root;
    }
    
    private ListNode findMid(ListNode head){
      ListNode fast = head;
      ListNode slow = head;
      
      while(fast.next != null && fast.next.next != null){
        fast = fast.next.next;
        slow = slow.next;
      }
      return slow;
    }
}

 

109. Convert Sorted List to Binary Search Tree

标签:while   tail   nod   dmi   exception   lin   span   ase   list   

原文地址:https://www.cnblogs.com/tobeabetterpig/p/9490875.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!