数据结构与算法

LeetCode-109-有序链表转换二叉搜索树

2021-11-17  本文已影响0人  醉舞经阁半卷书

有序链表转换二叉搜索树

题目描述:给定一个单链表,其中的元素按升序排序,将其转换为高度平衡的二叉搜索树。

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例说明请见LeetCode官网。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一:递归

首先,将链表转化成数组,然后处理过程就和 LeetCode-108-将有序数组转换为二叉搜索树 的解法完全一样了。

根据二叉搜索树的性质,因为给定的数组是按升序排列的,所以可以确定数组num即为该二叉搜索树的中序遍历序列,为了得到一颗平衡的二叉树,取数组中间位置的节点作为根节点,这样,左右子树的节点较为平衡,具体处理过程如下:

  • 调用递归方法,初始的起始位置为数组的长度;
  • 当起点位置大于终点位置时,说明节点已经遍历完了,直接返回空树;
  • 获取中间位置的值作为根节点,这样左右子树的节点树是比较均衡的;
  • 然后递归获得当前根节点的左右子树;
  • 最后返回根节点即为平衡的二叉搜索树。
import com.kaesar.leetcode.ListNode;
import com.kaesar.leetcode.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

public class LeetCode_109 {
    public static TreeNode sortedListToBST(ListNode head) {
        Queue<Integer> queue = new LinkedList<>();
        while (head != null) {
            queue.add(head.val);
            head = head.next;
        }
        int[] nums = new int[queue.size()];
        int index = 0;
        // 先将链表转换成数组,即为二叉树的中序遍历序列
        while (!queue.isEmpty()) {
            nums[index++] = queue.poll();
        }
        // 调用递归方法,初始的起始位置为数组的长度
        return sortedListToBST(nums, 0, nums.length - 1);
    }

    /**
     * 递归
     *
     * @param nums
     * @param left
     * @param right
     * @return
     */
    private static TreeNode sortedListToBST(int[] nums, int left, int right) {
        // 当起点位置大于终点位置时,说明节点已经遍历完了,直接返回空树
        if (left > right) {
            return null;
        }
        // 获取中间位置的值作为根节点,这样左右子树的节点树是较为平衡点
        int mid = (left + right) / 2;
        TreeNode root = new TreeNode(nums[mid]);
        // 然后递归获得当前根节点的左右子树
        root.left = sortedListToBST(nums, left, mid - 1);
        root.right = sortedListToBST(nums, mid + 1, right);
        return root;
    }

    public static void main(String[] args) {
        ListNode head = new ListNode(-10);
        head.next = new ListNode(-3);
        head.next.next = new ListNode(-0);
        head.next.next.next = new ListNode(5);
        head.next.next.next.next = new ListNode(9);

        sortedListToBST(head).print();
    }
}

【每日寄语】 任铁任金,定有可穿之砚;日磨日削,从无不锐之针。

上一篇下一篇

猜你喜欢

热点阅读