Leetcode 108 將有序陣列轉換為二叉搜尋樹
阿新 • • 發佈:2020-12-01
Leetcode 108 將有序陣列轉換為二叉搜尋樹
資料結構定義:
將一個按照升序排列的有序陣列,轉換為一棵高度平衡二叉搜尋樹。 本題中,一個高度平衡二叉樹是指一個二叉樹每個節點的左右兩個子樹的高度差的絕對值不超過 1。 示例: 給定有序陣列: [-10,-3,0,5,9], 一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜尋樹: 0 / \ -3 9 / / -10 5 /** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */
遞迴寫法:
/* * 思路: 中序遍歷,選擇中間節點作為root,左邊的數字作為左子樹 */ class Solution { public TreeNode sortedArrayToBST(int[] nums) { if (nums == null || nums.length <= 0) { return null; } TreeNode root = dfs(nums, 0, nums.length - 1); return root; } private TreeNode dfs(int[] nums, int left, int right) { if (left > right) { return null; } int mid = left + ((right - left) >> 1); TreeNode root = new TreeNode(nums[mid]); root.left = dfs(nums, left, mid - 1); root.right = dfs(nums, mid + 1, right); return root; } }