求一个二叉树每一层节点个数

import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

public class Test4 {
    public static void main(String[] args) {

        /*
            A
           / \
          B   C
         /   / \
        D   E   F
             \
              G
        */
        Node A = new Node(null, null, "A");
        Node B = new Node(null, null, "B");
        Node C = new Node(null, null, "C");
        Node D = new Node(null, null, "D");
        Node E = new Node(null, null, "E");
        Node F = new Node(null, null, "F");
        Node G = new Node(null, null, "G");

        A.left = B;
        A.right = C;
        B.left = D;
        C.left = E;
        C.right = F;
        E.right = G;

        Map<Integer, Integer> res = getRes(A);
        System.out.println("树的层数:" + res.size());
        Set<Entry<Integer,Integer>> entrySet = res.entrySet();
        for (Entry<Integer, Integer> entry : entrySet) {
            System.out.println(entry.getKey() + " : " + entry.getValue());
        }
    }
    public static Map<Integer, Integer> getRes(Node node){
        Map<Integer, Integer> res = new HashMap<>();
        LinkedList<Node> currQueue = new LinkedList<>();
        LinkedList<Node> nextQueue = null;
        currQueue.addLast(node);
        int index = 1;
        int count = 1;
        res.put(index++, count);
        while(!currQueue.isEmpty()){
            nextQueue = new LinkedList<>();
            count = 0;
            while(!currQueue.isEmpty()){
                Node n = currQueue.getFirst();
                currQueue.removeFirst();
                if(n.left != null){
                    count++;
                    nextQueue.addLast(n.left);
                }
                if(n.right != null){
                    count++;
                    nextQueue.addLast(n.right);
                }
            }
            if(count > 0){
                res.put(index++, count);
            }
            currQueue = nextQueue;
        }
        return res;
    }
}
class Node{
    Node left;
    Node right;
    String data;
    public Node(Node left, Node right, String data) {
        super();
        this.left = left;
        this.right = right;
        this.data = data;
    }
}
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值