美团笔试第一题

思路:
N个节点N-1条边的连通图一定是树,不存在环路,
所以根节点如果有多个子树,遍历完一个子树之后必须回到根节点才能进入其它子树,
只有最后一个子树不必返回,所以应该把最长的子树放在最后访问
递归表达式为:
dfs(start) = sum(dfs(i)+2) - depth
其中i表示第i个子树,最后一个子树不必返回,所以减去树的高度

import java.util.Scanner;

public class Meituan1 {
    static int N;
    static int[][] mat;
    static boolean[] flag; //标记节点是否已访问过

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        mat = new int[N][N];
        int n = N;
        while (n-- > 1) {
            int i = sc.nextInt();
            int j = sc.nextInt();
            mat[i - 1][j - 1] = mat[j - 1][i - 1] = 1;
        }
        flag = new boolean[N];
        int maxDepth = depth(0);
        flag = new boolean[N];
        System.out.println(dfs(0) - maxDepth);
        sc.close();
    }
    
     //递归求解访问所有子树并返回的路径总和
    static int dfs(int start) {
        flag[start] = true;
        int res = 0;
        for (int i = 0; i < N; i++) {
            if (i != start && mat[start][i] != 0 && !flag[i]) {
                res += 2 + dfs(i);
            }
        }
        return res;
    }
    
     //递归求解树的高度
    static int depth(int start) {
        flag[start] = true;
        int res = 0;
        for (int i = 0; i < N; i++) {
            if (i != start && mat[start][i] != 0 && !flag[i]) {
                int idepth = 1 + depth(i);
                if (idepth > res) res = idepth;
            }
        }
        return res;
    }
}
然而只过了9%。。。求大佬指正。。
#美团##笔试题目##题解#
全部评论
这道题的题目你还记得吗。。。我想自己做一下谢谢了
点赞 回复 分享
发布于 2018-09-26 14:22
1. N最大是100000,所以不能用邻接矩阵,必须用邻接表,否则时空都爆炸 2. 算法就有问题,实际上每条边刚好访问2次,除了最长的那条路径不用回去,所以答案是2*(N-1)-maxd,考虑到N很大有可能爆栈,所以你还要用非递归的方法算maxd。
点赞 回复 分享
发布于 2018-09-07 14:50

相关推荐

不愿透露姓名的神秘牛友
07-08 13:15
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-09 12:10
直接上图
牛客13578115...:改得一般,不值80
点赞 评论 收藏
分享
头顶尖尖的程序员:我也是面了三四次才放平心态的。准备好自我介绍,不一定要背熟,可以记事本写下来读。全程控制语速,所有问题都先思考几秒,不要急着答,不要打断面试官说话。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务