栈和队列 构造数组的MaxTree

【题目】
定义二叉树如下:

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());
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值