1. 程式人生 > >LeetCode之根據序列求平衡二叉樹

LeetCode之根據序列求平衡二叉樹

問題描述:

/**
 * Given an array where elements are sorted in ascending order, convert it to a height balanced BST.
 */

給你一個升序的序列,根據這個序列來構造一個平衡二叉樹。
解題思路就是先找到根節點,然後遞迴找到左右子節點。
程式碼如下:

public static TreeNode sortedArrayToBST(int[] num) {
        return sortedArrayToBST(num, 0, num.length - 1);
    }

    public
static TreeNode sortedArrayToBST(int[] num, int start, int end) { if (start <= end) { int mid = (start + end) / 2; TreeNode left = sortedArrayToBST(num, start, mid - 1); TreeNode right = sortedArrayToBST(num, mid + 1, end); TreeNode node = new
TreeNode(num[mid]); node.left = left; node.right = right; return node; } return null; }

遞迴求子節點時,跟二分查詢類似。