1004 Counting Leaves（Java）

本文讲一下PAT题库里面的1004 Counting Leaves

题目

解释

第一行给出两个整数，M代表一共多少个节点，N代表下面输出几行也就是非叶子结点的数量

第一个数是非叶子结点的编号，第二个数是其非叶子节点孩子的数量，后面是它的孩子的编号

问我们从根节点开始，也就是01节点开始，一行一行遍历过去（树的层序遍历），输出每一行叶子结点的数量，注意格式

例如

解题思路 

我们用ArrayList[]保存每一个节点的孩子，从根节点开始，每遍历一次孩子，就保存其孙子，如果没有孙子说明是叶子结点，res++；，每层序遍历一次，就输出一次，注意格式！

代码

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static int M;
    public static ArrayList<Integer>[] list;

    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        M=sc.nextInt();
        if (M==1){
            System.out.println(1);
        }else{
            list=new ArrayList[M+1];
            int N=sc.nextInt();
            for (int i = 0; i < M + 1; i++) {
                list[i]=new ArrayList<>();
            }
            for (int i = 0; i < N; i++) {
                int root=Integer.parseInt(sc.next());
                int child=sc.nextInt();
                for (int j = 0; j < child; j++) {
                    list[root].add(Integer.parseInt(sc.next()));
                }
            }
            if (list[1].size()==0){
                System.out.println(1);
            }else{
                System.out.print(0);
            }
            int res=0;
            sout(list[1]);
        }


    }
    public static void sout(ArrayList<Integer> list1){
        int res=0;
        ArrayList<Integer> child=new ArrayList<>();
        for (Integer i : list1) {
            if (list[i].size()==0){
                res++;
            }else{
                child.addAll(list[i]);
            }
        }
        System.out.print(" "+res);
        if (child.size()!=0){
            sout(child);
        }
    }
}