滴滴9.13笔试第二题,有哪位大佬告诉我错哪了吗,提交爆0

import java.util.*;

public class T2 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int T = sc.nextInt();
        String[] res = new String[T];
        for(int i=0;i<T;i++){
            int n = sc.nextInt(), m = sc.nextInt(), k = sc.nextInt();
            Map<Integer,List<Integer>> adj = new HashMap<>();
            for(int j=0;j<m;j++){
                int a = sc.nextInt(), b = sc.nextInt(), v = sc.nextInt();
                if(v>k){continue;}
                if(!adj.containsKey(a)){
                    adj.put(a,new ArrayList<>());
                }
                if(!adj.containsKey(b)){
                    adj.put(b,new ArrayList<>());
                }
                adj.get(a).add(b);
                adj.get(b).add(a);
            }
            if(bfs(adj,n)){
//                res[i] = "YES";
                System.out.println("YES");
            }else{
//                res[i] = "NO";
                System.out.println("NO");
            }
        }
    }
    private static boolean bfs(Map<Integer,List<Integer>> adj, int n){
        if(adj.size()==0){return false;}
        int cnt = 0;
        Queue<Integer> queue = new LinkedList<>();
        Set<Integer> visited = new HashSet<>();
        int root = adj.entrySet().iterator().next().getKey();
        queue.offer(root);
        visited.add(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i=0;i<size;i++){
                int cur = queue.poll();
                cnt++;
                for(int next: adj.get(cur)){
                    if(!visited.contains(next)){
                        queue.offer(next);
                        visited.add(next);
                    }
                }
            }
        }
        return cnt==n?true:false;
    }
}

#笔试题目#
全部评论
自顶,真的感谢!!!
点赞 回复 分享
发布于 2020-09-13 20:47
AK C++题解: https://www.nowcoder.com/discuss/509200?type=post&order=create&pos=&page=0&channel=666&source_id=search_post
点赞 回复 分享
发布于 2020-09-13 20:57

相关推荐

jack_miller:杜:你不用我那你就用我的美赞臣
点赞 评论 收藏
分享
头像
10-22 19:18
上海大学 后端
jopajhhdjwnqk:水印都叠杀人书了
点赞 评论 收藏
分享
1 1 评论
分享
牛客网
牛客企业服务