今天看到微信上九章算法发了一个题目,貌似不是很难的样子:
给定一个没有重复元素的数组A,定义A上的MaxTree为:MaxTree的根节点为A中最大的数,根节点的左子树为数组中最大数左边部分的MaxTree,右子树为数组中最大数右边部分的MaxTree,给出一个数组A,构造这个数组的MaxTree。
读完一遍题目就很容易想到分治法:
struct Node{
int val;
Node* left;
Node* right;
Node(int v): val(v), left(NULL), right(NULL){}
};
Node* buildMaxTree(int arr[], int l, int r)
{
if(l == r) return new Node(arr[l]);
int m = max_element(arr + l, arr + r + 1) - arr;
Node* root = new Node(arr[m]);
if(m > l) root->left = buildMaxTree(arr, l, m - 1);
if(m < r) root->right = buildMaxTree(arr, m + 1, r);
return root;
}
可以看到建树的过程类似快排,都是先遍历一遍数组,再将数组划分为两个子数组分别操作,所以也容易想到时间复杂度平均情况和最好情况下为O(nlogn),最差情况(同快排一样都是数组已经是排序好的情况下)是O(n^2),空间复杂度平均和最好情况是O(logn),最差是O(n)。
与快排不同的是,建树过程是找到子数组中的最大值,这个操作需要遍