LeetCode之根據序列求平衡二叉樹
阿新 • • 發佈:2019-02-19
問題描述:
/**
* 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;
}
遞迴求子節點時,跟二分查詢類似。