构造数组的MaxTree

构造数组的MaxTree

给定一个数组，数组的MaxTree定义如下：
1.数组必须没有重复元素
2.MaxTree是一个二叉树,数组的每一个值对应一个二叉树节点
3.MaxTree及其子树,值最大的节点都是数的根节点

思路

经过分析可以得到两个原则:
1.每个节点的父节点是它左边第一个比它大的数和它右边第一个比它大的数中,较小的那个
2.如果一个数左右两边都没有,这个数就是整个树的根节点
求左右两边第一个比它大的数,可以使用栈结构

代码

package Main;

import java.util.HashMap;
import java.util.Stack;

public class Main {

	public static class Node {
		public int value;
		public Node left;
		public Node right;

		public Node(int val) {
			this.value = val;
		}
	}

	public static Node getMaxTree(int[] arr) {
		Node[] nArr = new Node[arr.length];
		for (int i = 0; i != arr.length; i++) {
			nArr[i] = new Node(arr[i]); // 用数值构造出节点并保存在数组中
		}
		Stack<Node> stack = new Stack<>();
		HashMap<Node, Node> lBigMap = new HashMap<>();  // 用来保存每个节点左边数值比自己大的第一个节点
		HashMap<Node, Node> rBigMap = new HashMap<>();  // 右边
		for (int i = 0; i < nArr.length; i++) {
			Node curNode = nArr[i];
			while (!stack.isEmpty() && stack.peek().value < curNode.value) {  //如果栈顶节点比自身节点的数值大，则从栈中弹出，同时构造弹出节点的lBigMap
				popStackSetMap(stack, lBigMap);
			}
			stack.push(curNode);  //栈顶结点为空或者比自己节点数值大，入栈
		}
		while (!stack.empty()) {
			popStackSetMap(stack, lBigMap);  
		}
		for (int i = nArr.length - 1; i >= 0; i--) {   //从右边开始，找右边的第一个比自己数值大的节点
			Node curNode = nArr[i];
			while (!stack.isEmpty() && stack.peek().value < curNode.value) {
				popStackSetMap(stack, rBigMap);
			}
			stack.push(curNode);
		}
		while (!stack.isEmpty()) {
			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 static void popStackSetMap(Stack<Node> stack, HashMap<Node, Node> map) {
		Node popNode = stack.pop();
		if (stack.isEmpty()) {
			map.put(popNode, null);
		} else {
			map.put(popNode, stack.peek());
		}
	}
}

这是从左程云左神的书上所学的知识的总结。才疏学浅，若有错误欢迎大家多多提出，定虚心接受。