题目:有序链表转换二叉搜索树 middle
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
package leetCode.DFS;
import sun.reflect.generics.tree.Tree;
import java.util.LinkedList;
public class lc_dfs_109_sortedListToBST {
/*
题目:有序链表转换二叉搜索树 middle
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
*/
/*
思路:
本题和108题一样的思路,只不过不能直接使用数组的下标,需要找到链表的中间结点。
1)将其转化为数组,即将ListNode存入list中,当做数组处理,就和上题的步骤基本一样
2)利用双指针(快慢指针)来访问链表中的中间元素
本题采用快慢指针
具体步骤:
1.由于我们得到的是一个有序链表而不是数组,我们不能直接使用下标来访问元素。我们需要知道链表中的中间元素。
2.我们可以利用两个指针来访问链表中的中间元素。假设我们有两个指针 slow 和 fast。
slow 每次向后移动一个节点而 fast 每次移动两个节点。
当 fast 到链表的末尾时 slow 就访问到链表的中间元素。对于一个偶数长度的数组,
中间两个元素都可用来作二叉搜索树的根。
3.当找到链表中的中间元素后,我们将链表从中间元素的左侧断开,
做法是使用一个 prev 的指针记录 slow之前的元素,也就是满足 prev.next = slow。
断开左侧部分就是让 prev.next = None。
我们只需要将链表的头指针传递给转换函数,进行高度平衡二叉搜索树的转换。
所以递归调用的时候,左半部分我们传递原始的头指针;右半部分传递 slow.next 作为头指针。
亮点:
一开始我的做法是要使用head和tail两个结点来保证双指针在找mid时不会越界,
即helper(ListNode head,ListNode tail),递归结束条件:
if(head.val>tail.val || tail==null || head==null)为true时,返回null,
且fast走两步的判断条件也比较复杂,而且需要在第一次使用helper时,遍历一次head得到tail,效率低。
步骤3中直接将pre结点的prev.next = null,这样保证了使用双指针在找mid不会越界,直接判断null即可,
即保证head为null或者head.next=null即可。在fast走两步时也只需要保证
while(fast != null && fast.next != null)皆可,非常方便
*/
public static void main(String[] args) {
ListNode node1 = new ListNode(-10);
ListNode node2 = new ListNode(-3);
ListNode node3 = new ListNode(0);
ListNode node4 = new ListNode(5);
ListNode node5 = new ListNode(9);
node1.next = node2;
node2.next = node3;
node3.next = node4;
node4.next = node5;
lc_dfs_109_sortedListToBST m = new lc_dfs_109_sortedListToBST();
m.sortedListToBST(node1);
}
public TreeNode sortedListToBST(ListNode head) {
if (head == null)
return null;
return helper(head);
}
public TreeNode helper(ListNode head) {
//空结点直接返回null
if (head == null)
return null;
//!!!! 此处一定要判断,只有一个结点时,直接返回该结点,
// 否则head会不断进行递归,死循环,要注意跳出的条件
if (head.next == null)
return new TreeNode(head.val);
ListNode fast = head;
ListNode slow = head;
ListNode pre = null;//保存slow前一个结点
while (fast != null && fast.next != null) {
fast = fast.next.next;
pre = slow;
slow = slow.next;
}
if (pre != null)//注意pre在只有一个结点时为null
pre.next = null;//!!!亮点,这个妙
TreeNode root = new TreeNode(slow.val);
root.left = helper(head);
root.right = helper(slow.next);
return root;
}
static class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
}
}
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
}