滴滴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;
}
}
#笔试题目#