二叉搜索树的中序遍历是升序序列,题目给定的数组是按照升序排序的有序数组,因此可以确保数组是二叉搜索树的中序遍历序列
给定二叉搜索树的中序遍历,是否可以唯一地确定二叉搜索树?答案是否定的。如果没有要求二叉搜索树的高度平衡,则任何一个数字都可以作为二叉搜索树的根节点,因此可能的二叉搜索树有多个
如果增加一个限制条件,即要求二叉搜索树的高度平衡,是否可以唯一地确定二叉搜索树?答案仍然是否定的
直观地看,我们可以选择中间数字作为二叉搜索树的根节点,这样分给左右子树的数字个数相同或只相差 1,可以使得树保持平衡。如果数组长度是奇数,则根节点的选择是唯一的,如果数组长度是偶数,则可以选择中间位置左边的数字作为根节点或者选择中间位置右边的数字作为根节点,选择不同的数字作为根节点则创建的平衡二叉搜索树也是不同的
方法一:中序遍历,总是选择中间位置左边的数字作为根节点
struct TreeNode
{
int val;
TreeNode* left;
TreeNode* right; TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
TreeNode(int x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {}
};
class Solution
{
public:
TreeNode* helper(vector<int>& nums, int left, int right)
{
if (left > right) return nullptr;
int mid = (left + right) / 2;
TreeNode* root = new TreeNode(nums[mid]);
root->left = helper(std::ref(nums), left, mid-1);
root->right = helper(std::ref(nums), mid + 1, right);
return root;
}
TreeNode* sortedArrayToBST(vector<int>& nums)
{
return helper(std::ref(nums), 0, nums.size() - 1);
}
};
int main()
{
Solution A;
vector<int> nums ={ -10,-3,0,5,9 };
TreeNode* root = A.sortedArrayToBST(std::ref(nums));
return 0;
}
时间复杂度:O(n),其中 n 是数组的长度。每个数字只访问一次
空间复杂度:O(logn),其中 n 是数组的长度。空间复杂度不考虑返回值,因此空间复杂度主要取决于递归栈的深度,递归栈的深度是 O(logn)
方法二:中序遍历,总是选择中间位置右边的数字作为根节点
struct TreeNode
{
int val;
TreeNode* left;
TreeNode* right; TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
TreeNode(int x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {}
};
class Solution
{
public:
TreeNode* helper(vector<int>& nums, int left, int right)
{
if (left > right) return nullptr;
int mid = (left + right + 1) / 2;
TreeNode* root = new TreeNode(nums[mid]);
root->left = helper(std::ref(nums), left, mid-1);
root->right = helper(std::ref(nums), mid + 1, right);
return root;
}
TreeNode* sortedArrayToBST(vector<int>& nums)
{
return helper(std::ref(nums), 0, nums.size() - 1);
}
};
int main()
{
Solution A;
vector<int> nums ={ -10,-3,0,5,9 };
TreeNode* root = A.sortedArrayToBST(std::ref(nums));
return 0;
}
时间复杂度:O(n),其中 n 是数组的长度。每个数字只访问一次。
空间复杂度:O(logn),其中 n 是数组的长度。空间复杂度不考虑返回值,因此空间复杂度主要取决于递归栈的深度,递归栈的深度是 O(logn)