文章目录
- 一、题目
- 二、解法
- 三、完整代码
所有的LeetCode题解索引,可以看这篇文章——【算法和数据结构】LeetCode题解。
一、题目
二、解法
思路分析:这道题给我们的是一个有序数组,并要求构成一个平衡二叉搜索树,二叉搜索树的很容易理解,前几篇文章都做过类似的题目,又是平衡的,平衡就是说每个节点的左右子树高度差不超过1。对于这样一道题,我们可以用二分法将数组分为两个部分,以数组最中间的数为划分区间,这个分出来的两个区间元素个数差不会超过1,用递归的方式不断地划分出两个区间,中间的元素就作为根节点。程序当中我们将中间的元素作为根节点的值,然后用根节点的左右孩子去接收递归的返回值,终止条件又两种情况,一个元素的和两个元素的区间,一个元素的直接返回即可,又因为是有序数组,两个元素的在创建一个节点令它作为根节点的右孩子(二叉搜索树只能作为右孩子)。
程序如下:
class Solution {
public:TreeNode* traversal(vector<int>nums, int begin, int end) { int middle = (begin + end) / 2;TreeNode* root = new TreeNode(nums[middle]);if ((end - begin) <= 1) { // 两个元素或者一个元素if ((end - begin) == 1) { // 两个元素TreeNode* node = new TreeNode(nums[end]);root->right = node;}return root;}root->left = traversal(nums, begin, middle - 1); // 左子树,[begin, middle-1]左闭右闭 root->right = traversal(nums, middle + 1, end); // 右子树,[middle+1, end]左闭右闭return root;}TreeNode* sortedArrayToBST(vector<int>& nums) {return traversal(nums, 0, nums.size() - 1);}
};
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n)。
- 空间复杂度: O ( n ) O(n) O(n)。
三、完整代码
# include <iostream>
# include <vector>
# include <string>
# include <queue>
using namespace std;// 树节点定义
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* traversal(vector<int>nums, int begin, int end) { int middle = (begin + end) / 2;TreeNode* root = new TreeNode(nums[middle]);if ((end - begin) <= 1) { // 两个元素或者一个元素if ((end - begin) == 1) { // 两个元素TreeNode* node = new TreeNode(nums[end]);root->right = node;}return root;}root->left = traversal(nums, begin, middle - 1); // 左子树,[begin, middle-1]左闭右闭 root->right = traversal(nums, middle + 1, end); // 右子树,[middle+1, end]左闭右闭return root;}TreeNode* sortedArrayToBST(vector<int>& nums) {return traversal(nums, 0, nums.size() - 1);}
};template<typename T>
void my_print(T& v, const string msg)
{cout << msg << endl;for (class T::iterator it = v.begin(); it != v.end(); it++) {cout << *it << ' ';}cout << endl;
}template<class T1, class T2>
void my_print2(T1& v, const string str) {cout << str << endl;for (class T1::iterator vit = v.begin(); vit < v.end(); ++vit) {for (class T2::iterator it = (*vit).begin(); it < (*vit).end(); ++it) {cout << *it << ' ';}cout << endl;}
}// 层序遍历
vector<vector<int>> levelOrder(TreeNode* root) {queue<TreeNode*> que;if (root != NULL) que.push(root);vector<vector<int>> result;while (!que.empty()) {int size = que.size(); // size必须固定, que.size()是不断变化的vector<int> vec;for (int i = 0; i < size; ++i) {TreeNode* node = que.front();que.pop();vec.push_back(node->val);if (node->left) que.push(node->left);if (node->right) que.push(node->right);}result.push_back(vec);}return result;
}int main()
{vector<int> nums = { -10, -3, 0, 5, 9 };my_print(nums, "目标树");Solution s;TreeNode* root = s.sortedArrayToBST(nums);vector<vector<int>> tree = levelOrder(root);my_print2<vector<vector<int>>, vector<int>>(tree, "目标树:");system("pause");return 0;
}
end