【遇见时光】小米笔试-树的高度-java

思路：类似于层次遍历，用队列实现。每层结点进队列，末尾加入-1；再出队列，同时添加结点的子节点入队列，遇到-1则深度加1；

java代码：

package test;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
class Node{
    
    public Node(int father, int child) {
        this.father = father;
        this.child = child;
    }
    int father;
    int child;
}

public class tree {
    /****
     * 层次遍历的思想，队列实现，按层添加到队列，末尾入队-1；
     */
    public static int deep(int n,ArrayList<Node> tree){
        int count=0;
        int j=0;//记录操作了多少个结点；
        if(n<=0){
            return 0;
        }else if(n==1){
            return 2;
        }else{
            Queue<Integer> queue=new LinkedList<Integer>();
            queue.offer(0);
            j++;
            queue.offer(-1);//
            boolean flag=false;//判断是否到尾结点；
            while(!queue.isEmpty()){
//　　　　　　　　printdata(queue);
                Integer data=queue.poll();
//                System.out.println();
//                System.out.println("data:"+data);
//                System.out.println();
                if(j==n&&flag==false){
                    queue.offer(-1);
                    flag=true;
                }else if(data==-1){    
                    if(flag){
                        count++;
                    }else{
                        queue.offer(-1);
                        count++;
                    }                    
                }else{
                    for(int i = 0;i < tree.size(); i ++){
                        if(tree.get(i).father==data){
                            queue.offer(tree.get(i).child);
                            j++;
                        }
                    }                    
                }
            }
            
        }

        return count;
    }
    public static void printdata(Queue<Integer> queue){
        for(Iterator it2 = queue.iterator();it2.hasNext();){
            System.out.println("Q:"+it2.next());
       }
    }

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner in =new Scanner(System.in);
//        while(in.hasNext()){
            int n=in.nextInt();
            ArrayList<Node> tree =new ArrayList<Node>();
            for(int i=0;i<n-1;i++){
                int father=in.nextInt();
                int child=in.nextInt();
                Node node=new Node(father,child);
                tree.add(node);            }
            System.out.println(deep(n,tree));
//        }
    }

}
/**
 * 
 * 
9
0 1
0 2
1 3
1 4
2 5
2 6
3 7
3 8

5
0 1
0 2
1 3
1 4
**/

运行结果：

9
0 1
0 2
1 3
1 4
2 5
2 6
3 7
3 8
4

没有在赛码网ac过，大家可以试试。