【题目】
定义二叉树如下:
public class Node{
public int value;
public Node left;
public Node right;
public Node(int data){
this.value=value;
}
}
一个数组的MaxTree定义如下:
◆ 数组必须没有重复元素
◆ MaxTree是一颗二叉树,数组的每一个值对应一个二叉树节点
◆ 包括MaxTree树在内且在其中的每一颗子树上,值最大的节点都是树的头。
给定一个没有重复元素arr,写出生成这个数组的MaxTree函数,要求如果数组长度为N,则时间复杂度为O(N),额外空间复杂度为O(N)。
【解答】
以以下原则来建议这棵树:
◆ 每一个数的父节点是它左边第一个比他大的数和右边第一个比他大的数中,较小的那个。
◆ 如果一个数左边没有比它大的数,右边也没有,也就是说,这个数是整个数组的最大值,那么这个数就是MaxTree的头节点。
为什么通过这个方法构造?
1.通过这个方法,所有的数都能生成一棵树,这棵树可能不是二叉树,但肯定是一棵树,而不是一个森林。
2.通过这个方法,所有的数最多只有两个孩子,也就是说,这棵树可以用二叉树来表述,不用多叉树。
import java.util.HashMap;
import java.util.Map;
import java.util.Stack;
class Node{
public int value;
public Node left;
public Node right;
Node(int value){
this.value = value;
}
}
public class MaxTreeDemo {
public Node getMaxTree(int[] arr){
Node[] nArr = new Node[arr.length];
for (int i = 0; i != nArr.length; i++) {
nArr[i] = new Node(arr[i]);
}
Stack<Node> stack = new Stack<>();
Map<Node,Node> lBigMap = new HashMap<>();
Map<Node,Node> rBigMap = new HashMap<>();
for (int i = 0; i != nArr.length; i++) {
Node curNode = nArr[i];
while((!stack.empty())&&stack.peek().value<curNode.value)
popStackSetMap(stack,lBigMap);
stack.push(curNode);
}
while(!stack.empty())
popStackSetMap(stack, lBigMap);
for (int i = nArr.length-1; i != -1; i--) {
Node curNode = nArr[i];
while((!stack.empty())&&stack.peek().value<curNode.value)
popStackSetMap(stack,rBigMap);
stack.push(curNode);
}
while(!stack.empty())
popStackSetMap(stack, rBigMap);
Node head = null;
for(int i = 0; i != nArr.length; i++){
Node curNode = nArr[i];
Node left = lBigMap.get(curNode);
Node right = rBigMap.get(curNode);
if(left == null && right == null)
head = curNode;
else if(left == null){
if(right.left == null)
right.left = curNode;
else
right.right = curNode;
}
else if(right == null){
if(left.left == null)
left.left = curNode;
else
left.right = curNode;
}
else{
Node parent = left.value < right.value? left : right;
if(parent.left == null)
parent.left = curNode;
else
parent.right = curNode;
}
}
return head;
}
public void popStackSetMap(Stack<Node>stack,Map<Node,Node>map){
Node popNode = stack.pop();
if(stack.empty())
map.put(popNode, null);
else
map.put(popNode,stack.peek());
}
}