把排序数组转化为高度最小的二叉树

最新推荐文章于 2022-01-06 10:00:43 发布

原创

最新推荐文章于 2022-01-06 10:00:43 发布 · 859 阅读

1 ·

CC 4.0 BY-SA版权

通过将排序数组的中间元素作为根节点，并递归地将剩余元素分为左右子数组，可以构建高度最小的二叉树。这种方法确保每个根节点的左右子树高度差不超过1，满足平衡二叉树的条件。

把排序数组转化为高度最小的二叉树

思路：直接采用中间值来作为二叉树的根节点；将原数组分成左右均等或者相差一个数的两个新数组；然后递归的对这两个新数组进行相同的处理，这样对于每一个根节点，其左右子树的高度相差绝对值不会超过1，也就是满足了二叉平衡树的要求了。

struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int x) :val(x) {}
};

class Solution {
public:
	TreeNode* sortedArrToBST(vector < int>& array) {
		if (array.empty())return nullptr;
		return sortedArrToBSTcore(array, 0, array.size() - 1);
	}

	TreeNode* sortedArrToBSTcore(vector < int>& array, int low, int high) {
		if (low <= high) {
			int mid = (low + high) / 2;
			TreeNode* pRoot = new TreeNode(array[mid]);
			pRoot->