問題描述
將一個按照升序排列的有序數組,轉換爲一棵高度平衡二叉搜索樹。
本題中,一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。
示例:
給定有序數組: [-10,-3,0,5,9],
一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜索樹:
0
/ \
-3 9
/ /
-10 5
Java 實現
既然數組是有序的,構建平衡二叉樹時利用二分思想就很好實現了。樹的構建通常採用遞歸使用子樹根節點進行調用,對每次遞歸對數組從中間進行二分即可。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode sortedArrayToBST(int[] nums) {
return solve(nums, 0, nums.length - 1);
}
public TreeNode solve(int[] nums, int left, int right) {
if (left > right) {
return null;
}
int mid = (left + right + 1) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = solve(nums, left, mid - 1);
root.right = solve(nums, mid + 1, right);
return root;
}
}
鏈接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree/