将有序数组转换为二叉搜索树(C++)

108. 将有序数组转换为二叉搜索树

给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 1:
将有序数组转换为二叉搜索树(C++)_第1张图片
输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案

思路:每次用数组中间元素去建立一个新的节点,然后root->left等于数组左半部分区间的中间值,root->right等于数组右半部分的中间值。最后返回root。

代码:

class Solution {	//108. 将有序数组转换为二叉搜索树
public:
	TreeNode* traversal(vector<int>& nums, int left, int right) {
		if (left > right) return nullptr;
		int mid = left + ((right - left) >> 1);
		TreeNode* root = new TreeNode(nums[mid]);
		root->left = traversal(nums, left, mid - 1);
		root->right = traversal(nums, mid + 1, right);
		return root;
	}
	TreeNode* sortedArrayToBST(vector<int>& nums) {
		TreeNode* root = traversal(nums, 0, nums.size() - 1);
		return root;
	}
};

参考资料:代码随想录

你可能感兴趣的:(leetcode题,c++,leetcode,开发语言)