美团笔试第一题

思路:
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%。。。求大佬指正。。
#美团##笔试题目##题解#
全部评论
1. N最大是100000,所以不能用邻接矩阵,必须用邻接表,否则时空都爆炸 2. 算法就有问题,实际上每条边刚好访问2次,除了最长的那条路径不用回去,所以答案是2*(N-1)-maxd,考虑到N很大有可能爆栈,所以你还要用非递归的方法算maxd。
点赞 回复 分享
发布于 2018-09-07 14:50
这道题的题目你还记得吗。。。我想自己做一下谢谢了
点赞 回复 分享
发布于 2018-09-26 14:22

相关推荐

头发暂时没有的KFC总裁:找廉价劳动力罢了
点赞 评论 收藏
分享
起名字真难233:人家只有找猴子的预算,来个齐天大圣他们驾驭不住呀😂😂
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务