問題描述:
將一個按照升序排列的有序數組,轉換為一棵高度平衡二叉搜索樹。
本題中,一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。
示例:
給定有序數組: [-10,-3,0,5,9], 一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜索樹: 0 / \ -3 9 / / -10 5
解題思路:
用數組最中間的的值將數組分為三部分:左半部分、中間值、右半部分;
用中間值構造節點,數組左半部分構造節點的左子樹,右半部分右子樹;
對數組的左半部分與右半部分重復上述操作。
實現代碼:
private static TreeNode test(int nums[], int lo, int hi) { if (lo > hi) return null; if (lo == hi) return new TreeNode(nums[lo]); int mid = lo + (hi - lo) / 2; TreeNode node = new TreeNode(nums[mid]); node.left = test(nums, lo, mid-1); node.right = test(nums, mid+1, hi); return node; } public static TreeNode sortedArrayToBST(int nums[]) { return test(nums, 0, nums.length-1); }