#include <iostream>
#include <vector>
using namespace std;
/*
题目要求:
Given an array where elements are sorted in ascending order,
convert it to a height balanced BST.
二叉查找树的特点是,左子树比根小,根比右子树小
所以每次可以找有序数组的中间的数字作为根,左儿子是左子树的根,右儿子是右子树
的根,这样递归建树,最后返回根。
不过不用自己写内存管理?
*/
/**
* Definition for a binary tree node.
*/
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 buildTree(nums, 0, int(nums.size()) - 1);
}
private:
TreeNode* buildTree(vector<int>& array, int left, int right) {
if (left <= right) {
int mid = (left + right) / 2;
TreeNode* root = new TreeNode(array[mid]);
root->left = buildTree(array, left, mid - 1);
root->right = buildTree(array, mid + 1, right);
return root;
}
return NULL;
}
};
int main(int argc, const char * argv[]) {
vector<int> n = {1,2,3};
Solution s;
TreeNode* r = s.sortedArrayToBST(n);
return 0;
}
剑指offer里面有个面试题是说二叉树转双向链表的。到时做做。。