根据数组层次遍历构建二叉树

本文介绍了一个使用Java实现的二叉树数据结构,并通过层次遍历的方式获取指定层级的节点值,展示了如何构建二叉树、进行层次遍历以及输出特定范围内的层级数据。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

/**
    15
    15 7 20 -1 -1 3 12 -1 -1 -1 -1 14 8 -1 -1
    2 4
*/
import java.util.*;

public class Main2 {
    static int[]  value ;
    static class TreeNode{
        int val;
        TreeNode left,right;
        public TreeNode(int t) {
            this.val =t;
        }
    }
    static ArrayList<ArrayList<Integer>> rs;
    public static void main(String[] args) {
        Scanner cin = new Scanner(System.in);
        int n = cin.nextInt();
         value = new int[n];
        for(int i = 0;i < n;i++) {
             value[i] = cin.nextInt();
        }
        int s = cin.nextInt();
        int e = cin.nextInt();
        TreeNode head = getTree();
        getRs(head,s,e);
        for(ArrayList list : rs) {
            for(int i = 0;i < list.size();i++) {
                if(i == list.size() -1) {
                    System.out.println(list.get(i));
                } else {
                    System.out.print(list.get(i) +" ");
                }
            }
        }
    }



    private static void getRs(TreeNode root, int s, int e) {
        rs = new ArrayList();
        LinkedList<TreeNode> queue= new LinkedList();
        if(root == null) {
            return;
        }
        queue.offer(root);
        int ceng = 1;
        while(queue.size() > 0) {
            int size=queue.size();
            ArrayList<Integer> list=new ArrayList<>();

            for (int i = 0; i < size; i++) {
                TreeNode top = queue.poll();
                if (top == null || top.val == -1) {
                    continue;
                }
                list.add(top.val);
                queue.add(top.left);
                queue.add(top.right);
            }
            if(ceng >= s && ceng <= e) {
                rs.add(new ArrayList<>(list));
            }

            ceng++;
        }
    }


    private static TreeNode getTree(){
        TreeNode  p = new TreeNode(value[0]);
        TreeNode  q = p;
        Queue<TreeNode > queue = new LinkedList();
        int i = 0;
        while (p != null) {
            if(2*i+1<value.length){
                p.left = new TreeNode(value[2 * i + 1]);
                queue.add(p.left);
            }
            if (2*i+2<value.length){
                p.right = new TreeNode(value[2 * i + 2]);
                queue.add(p.right);
            }
            p = queue.poll();
            i++ ;
        }
        return q;
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值