数组变树

来源:互联网 发布:centos 6 php7 编辑:程序博客网 时间:2024/05/16 19:55

对于一个没有重复元素的整数数组,请用其中元素构造一棵MaxTree,MaxTree定义为一棵二叉树,其中的节点与数组元素一一对应,同时对于MaxTree的每棵子树,它的根的元素值为子树的最大值。现有一建树方法,对于数组中的每个元素,其在树中的父亲为数组中它左边比它大的第一个数和右边比它大的第一个数中更小的一个。若两边都不存在比它大的数,那么它就是树根。请设计O(n)的算法实现这个方法。

给定一个无重复元素的数组A和它的大小n,请返回一个数组,其中每个元素为原数组中对应位置元素在树中的父亲节点的编号,若为根则值为-1。

测试样例:
[3,1,4,2],4
返回:[2,0,-1,2]

class MaxTree {public:    vector<int> buildMaxTree(vector<int> A, int n) {       if(n == 0)  return A;        //初始状态下,所有元素的父结点都为NULL,对应下标为-1       vector<int> result(n, -1);//声明一个初始大小为n且初始值都为-1的向量       stack<int> sta;        //循环处理每一个元素        //需要一个额外的栈来辅助获取两个阶段的最大值(两个阶段指左边最大值和右边最大值)        //每次判断一个元素分三种情况        //1. 栈为空,也就是初始状态,直接将元素下标入栈(此时该元素的父结点下标为默认值-1)        //2. 栈不为空,同时即将进栈的元素比栈顶元素小,那么直接将元素进栈同时更新当前元素的        //   父结点下标为i(设定父结点为左边第一个比该元素大的,右侧第一个比该元素大的第3步说明)        //3. 栈不为空,同时即将进栈的元素比栈顶元素大,那么需要将栈顶元素出栈,然后分情况比较:        //     1) 栈顶元素对应父结点下标为NULL,说明栈顶元素左侧没有最大值,此时需要更新        //        栈顶元素对应父结点下标为当前进栈元素的下标i,然后将栈顶元素pop出栈        //     2) 栈顶元素对应的父结点下标对应数组元素比i位置对应数组元素大,由maxTree的        //        定义,必须保存左侧和右侧相对较小的那个作为最终结果,所以也更新元素下标为i,        //        然后将栈顶元素pop出栈        //     3) 如果不在上面两种情况范围内,直接将栈顶元素出栈,然后继续循环执行3步骤       for(int i=0;i<n;i++){            while(!sta.empty() && A[sta.top()]<A[i]){               // 如果即将被pop元素的左侧最大值为NULL或者其左侧最大值比右侧(A[i])最大值大,               // 需要先更新该元素的父结点为右侧结点下标                 if(result[sta.top()] == -1 || A[result[sta.top()]]>A[i]){                     result[sta.top()] = i;//更新成右侧最大值                 }                 sta.pop();//栈顶元素最终的父结点下标不管更新没更新,都要把其pop出栈             }             // 经过上面循环,判断一下栈是否为空了             if(!sta.empty()){//如果不空,更新当前元素左侧第一个比它大的父结点为栈顶元素                  result[i] = sta.top();             }             sta.push(i);        }        return result;    }};