algorithm - 将排序数组插入二叉搜索树

Question

我想实现一种将排序数组插入二叉搜索树的算法，但我不想以一棵只向一侧生长的树结束。

你有什么想法吗？

谢谢。

Answer 1

这应该给你一个平衡的树(在 O(n) 中):

1. 为数组中的中间元素构造一个节点并返回
   (这将是基本案例中的根)。
2. 从1.开始重复数组的左半部分，将返回值赋给根的左 child 。
3. 在数组的右半部分从1.开始重复，将返回值赋给根的右 child 。

类 Java 代码:

TreeNode sortedArrayToBST(int arr[], int start, int end) {
  if (start > end) return null;
  // same as (start+end)/2, avoids overflow.
  int mid = start + (end - start) / 2;
  TreeNode node = new TreeNode(arr[mid]);
  node.left = sortedArrayToBST(arr, start, mid-1);
  node.right = sortedArrayToBST(arr, mid+1, end);
  return node;
}

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

代码源自here .