阿里实习生笔试 2021/04/28 第二题记录

• 此题源于阿里2021/4/28 实习生笔试 第二题
• 给你一棵二叉搜索树（需要自己建树）
• 你返回树中左支或右支的最大长度 和 个数
  5
  3 6
  1 4 9
  2
  如上图所示 最大支为 5 3 1， 5 6 9

•  输入描述 ： n
  

•             5 3 6 1 2 9 4
  

•  输出描述 ： 3  2（最大长度为3  数量为 2个）
  

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

/*
* 此题源于阿里2021/4/28 实习生笔试 第二题
* 给你一棵二叉搜索树（需要自己建树）
* 你返回树中左支或右支的最大长度 和 个数
*        5
*    3       6
* 1     4       9
*  2
*      如上图所示 最大支为   5 3 1
*                            5 6 9
*      输入描述 ： n
*                 5 3 6 1 2 9 4
*      输出描述 ： 3  2（最大长度为3  数量为 2个）
* */
public class maxLeftOrRightLen {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        ArrayList<TreeNode> rec = new ArrayList<>(n);
        for(int i = 0;i < n; i ++){
            rec.add(new TreeNode(sc.nextInt()));
        }
        /* 验证记录表中的树节点顺序
        for(TreeNode temp : rec){
            System.out.println(temp.val);
        }*/
        TreeNode root = rec.get(0);

        for(int i = 1; i < n; i ++){
            myBuildTree(root,rec.get(i));
        }
       // System.out.println(root.left.val);
        //System.out.println(root.left.left.val);//验证是否建树成功
        int max = 0;
        HashMap<Integer,Integer> recMap = new HashMap<>();
        max = dfs(root,1,1,recMap,0);
        System.out.println(max + " " + recMap.get(max));

    }
    public static  class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;
        public  TreeNode(int val){
            this.val = val;
        }
    }
    public static TreeNode myBuildTree(TreeNode root,TreeNode child){
            if(root == null) return null;
            if(root.val > child.val){
                if(myBuildTree(root.left,child) == null)
                    root.left = child;
            }else{
                if(myBuildTree(root.right,child) == null)
                    root.right = child;
            }
            return  child;
    }
    public static  int dfs(TreeNode root, int lLen, int rLen, HashMap<Integer,Integer> recMap,int flag){
        if(root == null) return 0;
        int maxLeft = lLen;
        int maxRight = rLen;

        if(root.left != null){
            maxLeft = dfs(root.left,lLen ,1,recMap,1) + 1 ;
            //System.out.println(root.val +","+ maxLeft); 观察遍历情况
            recMap.put(maxLeft,recMap.getOrDefault(maxLeft,0) + 1);
        }if(root.right != null){
            maxRight = dfs(root.right,1,rLen  ,recMap,2) + 1;
            //System.out.println(root.val +","+ maxRight); 观察遍历情况
            recMap.put(maxRight,recMap.getOrDefault(maxRight,0) + 1);
        }
        //判断是左插入还是右插入，如果是根节点返回最大值
        if(maxLeft > maxRight && flag == 0){
            flag = 1;
        }else if(maxLeft < maxRight && flag == 0){
            flag = 2;
        }
        return flag == 1? maxLeft : maxRight;
    }
}