-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPrim.java
48 lines (45 loc) · 1.51 KB
/
Prim.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import java.lang.reflect.Array;
import java.util.*;
public class Prim {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int numv = in.nextInt();
in.nextLine();
int numr = in.nextInt();
in.nextLine();
List<List<List<Integer>>> grath = new ArrayList<>();
for (int i = 0; i < numv; i++) {
grath.add(new ArrayList<>());
}
for (int i = 0; i < numr; i++) {
var a = in.nextInt();
var b = in.nextInt();
var c = in.nextInt();
in.nextLine();
grath.get(a).add(List.of(b, c));
grath.get(b).add(List.of(a, c));
}
int sum = 0;
List<Integer> done = new ArrayList<>();
while (done.size() != numv) {
if (done.isEmpty()){
done.add(0);
}
int siz = Integer.MAX_VALUE;
int nv = 0;
for (int i = 0; i < done.size(); i++){
for (int j = 0; j < grath.get(done.get(i)).size(); j++){
if (!done.contains(grath.get(done.get(i)).get(j).get(0))){
if (grath.get(done.get(i)).get(j).get(1) <= siz){
siz = grath.get(done.get(i)).get(j).get(1);
nv = grath.get(done.get(i)).get(j).get(0);
}
}
}
}
sum += siz;
done.add(nv);
}
System.out.println(sum);
}
}