构造数组的MaxTree

给定一个没有重复元素的数组A，定义A上的MaxTree如下：MaxTree的根节点为A中最大的数，根节点的左子树为数组中最大数左边部分的MaxTree，右子树为数组中最大数右边部分的MaxTree。请根据给定的数组A，设计一个算法构造这个数组的MaxTree。

思路：

首先容易想到的是使用递归的方法来构造MaxTree，每一层递归用O(n)的时间找到最大数，然后将数组分为左右两个部分，然后递归完成构造。这种算法在极端情况下复杂度可能达到O(n^2)，所以并不能被面试官所接受。但是你首先至少要把这种暴力的方法答出，并分析出最坏时间复杂度。

如果能够确定每个节点的父亲节点，则可以构造出整棵树。找出每个数往左数第一个比他大的数和往右数第一个比他大的数，两者中较小的数即为该数的父亲节点。如：[3,1,2]，3没有父亲节点，1的父亲节点为2，2的父亲节为3。并且可以根据与父亲的位置关系来确定是左儿子还是右儿子。接下来的问题是如何快速找出每个数往左、往右第一个比他大的数。这里需要用到数据结构栈。以找每个数左边第一个比他大的数为例，从左到右遍历每个数，栈中保持递减序列，新来的数不停的Pop出栈顶直到栈顶比新数大或没有数。以[3,1,2]为例，首先3入栈，接下来1比3小，无需pop出3，1入栈，并且确定了1往左第一个比他大的数为3。接下来2比1大，1出栈，2比3小，2入栈。并且确定了2往左第一个比他大的数为3。用同样的方法可以求得每个数往右第一个比他大的数。时间复杂度O(n)，空间复杂度也是O(n)为最优解法。

这里最巧妙的一点是如何找某数的左边最近的比它大的值和右边最近的比它大的值。取左右两边的较小的数作为该数的父节点。

//
//  main.cpp
//  构造数组的MAXTree
//
//  Created by zjl on 16/6/2.
//  Copyright © 2016年 zjl. All rights reserved.
//

#include <iostream>
#include <vector>
#include <stack>
#include <map>
using namespace std;

struct Node{
    int val;
    Node* left;
    Node* right;
    Node(int data):val(data),left(NULL),right(NULL){}
};

//这是找数的左（右）边最近的比它大的数，并用map存起来
void  stackTomap(stack<Node*>& stk, map<Node*, Node*>& mapleft )
{
    Node* temp = stk.top();
    stk.pop();
    if(stk.empty())
        mapleft[temp] = NULL;
    else
        mapleft[temp] = stk.top();
}


Node* getMaxTree(vector<int> vec){
    vector<Node*> v;
    for(int i = 0; i < vec.size(); i++)
    {
        v.push_back(new Node(vec[i]));
    }
    stack<Node*>stk;
    map<Node*, Node*>mapleft, mapright;
    for(int i = 0; i < vec.size(); i++){
        Node* t = v[i];
        while(!stk.empty() && stk.top()->val < t->val)
        {
            stackTomap(stk, mapleft);
        }
        stk.push(t);
    }
    while(!stk.empty())
    {
        stackTomap(stk, mapleft);
    }
    
    for(int j = v.size()-1; j >= 0; j--)
    {
        Node* t = v[j];
        while(!stk.empty() && stk.top()->val < t->val)
            stackTomap(stk, mapright);
        stk.push(t);
    }
    while(!stk.empty())
        stackTomap(stk, mapright);
    Node* head = NULL;
    for(int k = 0; k < v.size(); k++){
        Node* current = v[k];
        Node* leftnode = mapleft[current];
        Node* rightnode = mapright[current];
        if(leftnode == NULL && rightnode == NULL)
            head = current;
        else if(leftnode == NULL){
            if(rightnode -> left == NULL)
                rightnode -> left = current;
            else
                rightnode -> right = current;
        }
        else if(rightnode == NULL){
            if(leftnode ->left == NULL)
                leftnode ->left = current;
            else
                leftnode ->right = current;
        }
        else{
            Node* parent = leftnode->val > rightnode->val? rightnode:leftnode;
            if(parent->left == NULL)
                parent->left = current;
            else
                parent->right = current;
        }
    }
    return head;

}

void print_tree(Node* root){
    if(root == NULL)
        return;
    print_tree(root->left);
    cout<< root->val<<" ";
    print_tree(root->right);
}

int main(int argc, const char * argv[]) {
    vector<int>vec = {3,5,4,1,2};
    Node* root = getMaxTree(vec);
    print_tree(root);
    return 0;
}

输入： 

3 5 4 1 2

输出：

3 5 1 2 4