2020-09-13 滴滴-2021校招在线笔试-DE数据开发试卷
1. D星群岛
题目:
D星群岛由n个小岛组成。为了加强小岛居民之间的交流,头目决定启动一个造桥工程,将全部n个岛连接到一起。
由于受到金融危机的影响,头目要求造桥的总成本要最少,并且还规定每一座桥的成本都不能超过k万D星币。
需要注意的是,由于受到地理环境和气候影响,有些小岛之间没有办法直接造桥。
现在给你m条小岛之间的造桥成本数据以及k的值,请问这个宏伟的造桥工程是否能够顺利完成?
注意:可能边不够,也可能费用超支。
输入描述
多组输入,第1行输入一个正整数T表示输入数据的组数。
对于每一组输入数据:输入m+1行。
第1行包含三个正整数,分别表示n、m和k,n≤100,m≤1000,k≤10000,三个数字之间用空格隔开。
接下来m行表示m条小岛之间的造桥成本数据,每一行包含三个正整数,分别表示两个小岛的编号(从1开始)和这两个小岛之间的造桥成本(单位:万)。
输出描述
针对每一组输入数据,如果能够完成造桥工程输出“Yes”,否则输出“No”。
样例输入
2
3 3 400
1 2 200
1 3 300
2 3 500
3 3 400
1 2 500
1 3 600
2 3 700
样例输出
Yes
No
代码:
dfs 判断连通图,dfs 一遍,采用一个 visited 数组剪枝,维护一个数用于记录遍历过的节点,当这个数等于所有小岛的个数说明该图是连通的。
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int T = in.nextInt();
for(int i = 0; i < T; i++) {
int n = in.nextInt();
int m = in.nextInt();
int k = in.nextInt();
map = new boolean[n+1][n+1];
visited = new boolean[n+1];
for(int j = 0; j < m; j++) {
int a = in.nextInt();
int b = in.nextInt();
int cost = in.nextInt();
if(cost <= k) {
map[a][b] = true;
map[b][a] = true;
}
}
dfs(1);
String res = count == n ? "Yes" : "No";
System.out.println(res);
count = 0;
}
}
public static int count = 0;
public static boolean[] visited;
public static boolean[][] map;
public static void dfs(int s) {
visited[s] = true;
count++;
// find next
for(int i = 1; i <= visited.length - 1; i+&