将有序数组转换为平衡二叉搜索树

问题

给定一个有序数组,数组元素升序排列,试将该数组转换为一棵平衡二叉搜索树(Balanced Binary Search Tree)。

思路

这个问题用递归很容易解出来。考虑下面一棵二叉搜索树:

这是一棵平衡的二叉搜索树,所谓平衡的定义,就是指二叉树的子树高度之差不能超过1。

如果要从一个有序数组中选择一个元素作为根结点,应该选择哪个元素呢?我们应该选择有序数组的中间元素作为根结点。

选择了中间元素作为根结点并创建后,剩下的元素分为两部分,可以看作是两个数组。这样剩下的元素在根结点左边的作为左子树,右边的作为右子树。、

解法

由上面的思路,我们可以在O(N)的时间内从有序数组创建一棵平衡的BST,使用分治算法,代码如下:

struct node* sortedArrayToBST(int arr[], int start, int end) 
{
  if (start > end) return NULL;
  // 这里同(start+left)/2,目的是为了防止溢出.
  int mid = start + (end - start) / 2;
  struct node *root = newNode(arr[mid]);//newNode创建二叉树结点,具体代码请看文章 二叉树问题汇总(1)
  root->left = sortedArrayToBST(arr, start, mid-1);
  root->right = sortedArrayToBST(arr, mid+1, end);
  return root;
}
 
struct node* sortedArrayToBST(int arr[], int n) 
{
  return sortedArrayToBST(arr, 0, n-1);
}

英文原文地址:http://www.leetcode.com/2010/11/convert-sorted-array-into-balanced.html





  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
将一个升序数组转换为高度平衡二叉搜索树,可以采用递归的方式来实现。 具体步骤如下: 1. 找到数组的中间元素,作为二叉搜索树节点。 2. 将数组分成左右两个子数组,左子数组中的元素都小于节点,右子数组中的元素都大于节点。 3. 递归地处理左右子数组,分别将它们转换为左右子树。 4. 将左右子树挂在节点下,构成一棵完整的二叉搜索树。 以下是代码示例: ```cpp #include <iostream> #include <vector> using namespace std; struct TreeNode { int val; TreeNode* left; TreeNode* right; TreeNode(int x) : val(x), left(NULL), right(NULL) {} }; class Solution { public: TreeNode* sortedArrayToBST(vector<int>& nums) { return buildBST(nums, 0, nums.size() - 1); } private: TreeNode* buildBST(vector<int>& nums, int left, int right) { if (left > right) { return NULL; } int mid = (left + right) / 2; TreeNode* root = new TreeNode(nums[mid]); root->left = buildBST(nums, left, mid - 1); root->right = buildBST(nums, mid + 1, right); return root; } }; int main() { vector<int> nums = {1, 2, 3, 4, 5, 6, 7}; Solution solution; TreeNode* root = solution.sortedArrayToBST(nums); // 遍历打印二叉树 return 0; } ``` 在这个示例中,我们定义了一个 `Solution` 类,其中包含了一个 `sortedArrayToBST` 函数,用于将升序数组转换为高度平衡二叉搜索树。 我们在 `sortedArrayToBST` 函数中调用了 `buildBST` 函数,用于递归地构建二叉搜索树。`buildBST` 函数的参数中,`nums` 表示原始数组, `left` 和 `right` 表示当前处理的数组区间。 在 `buildBST` 函数中,首先判断当前区间是否为空,如果为空则返回 `NULL`。然后计算出当前区间的中间位置 `mid`,将 `nums[mid]` 作为节点。再递归地处理左右子数组,分别将它们转换为左右子树。最后返回节点,构成一棵完整的二叉搜索树。 最后,我们可以使用遍历的方式打印出生成的二叉搜索树,以验证其正确性。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值