给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
把链表折半,中间的那个节点作根节点,左边的节点肯定是在左子树中的,右边的节点肯定是在右子树中的。然后继续递归下去。
Java代码实现,如若有误,欢迎指正,博主也是在学习的路上,大家共勉。
/**
* 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) {
//如果为空 返回null即可
if(head == null)
return null;
//如果链表只有一个节点,新建一个树节点返回即可
if(head.next == null)
return new TreeNode(head.val);
ListNode pre = head;//最前面的节点,用来断开链表的,下面会标注。
//定义两个链表节点,一个跑的慢,步长为1,一个跑的快,步长为2.这样子,当快的那个节点跑到链表结尾的时候,慢的那个节点刚好 跑到链表中间节点处
ListNode slow = pre.next;//慢节点
ListNode fast = slow.next;//快节点
while(fast != null && fast.next != null) {//注意这个判断条件
pre = slow;//维护中间节点的上一个节点
slow = slow.next;//慢节点一次步长为1
fast = fast.next.next;//快节点一次的步长为2
}
pre.next = null;//断开链表连接,pre是中间节点(slow)的上一个节点。
TreeNode root = new TreeNode(slow.val);//中间节点作为根节点
root.left = sortedListToBST(head);//从头结点到中间节点的上一个节点为止 作为左子树继续递归构造二叉搜索树
root.right = sortedListToBST(slow.next);//从中间节点的下一个节点开始到链表结尾为止作为右子树,继续递归构造二叉搜索树
return root;
}
}
运行结果