题目:
给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定的有序链表: [-10, -3, 0, 5, 9],
一个可能的答案是:[0, -3, 9, -10, null, 5], 它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
分析一:
一看到这个题目,脑子懵逼,两个字“不会”,建一个二叉搜索树是个什么东东。抓紧去看题解,找找别人解题的思路,豁然开朗啊,第一个思路就是递归了,方法巧的一。话归正传,题目让我们将单链表转换成发高度平衡的二叉搜索树,链表是升序排列的,学过数据结构的应该都清楚,平衡二叉排序树,值的大小应该是:左子树<=根节点<=右子树。因此,我们可以先找到这一个单链表的中间节点,并由此为二叉树的根节点,这时候左边就成为了左子树,右边就成为了右子树,是不是很熟悉这个形式,没错,这就符合了递归的条件,左右两棵子树就可以分别单独拉出来再次进行分割操作,直到只剩一个截点为止,退出。
代码:
/**
* 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 ListNode findMiddleElement(ListNode head) {
ListNode pre = null;
ListNode slow = head;
ListNode fast = head;
while(fast != null && fast.next != null) {
pre = slow;
slow = slow.next;
fast = fast.next.next;
}
if(pre != null) {
pre.next = null;
}
return slow;
}
public TreeNode sortedListToBST(ListNode head) {
if(head == null) {
return null;
}
ListNode mid = findMiddleElement(head);
TreeNode node = new TreeNode(mid.val);
if(head == mid) {
return node;
}
node.left = sortedListToBST(head);
node.right = sortedListToBST(mid.next);
return node;
}
}
分析二:
上面是直接用链表操作的,可是我们知道,当我们找中间节点的时候链表需要遍历整个链表,而数组却可以直接找到,因此,可以有现将链表转成数组,在进行下面的操作。
代码:
/**
* 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 findMiddleElement(List<Integer> list, int start, int end) {
if(start > end) {
return null;
}
int mid = (start + end) / 2;
TreeNode node = new TreeNode(list.get(mid));
if(start == end) {
return node;
}
node.left = findMiddleElement(list, start, mid - 1);
node.right = findMiddleElement(list, mid + 1, end);
return node;
}
public TreeNode sortedListToBST(ListNode head) {
List<Integer> list = new ArrayList<>();
ListNode pre = head;
while(pre != null) {
list.add(pre.val);
pre = pre.next;
}
return findMiddleElement(list, 0, list.size() - 1);
}
}