Repository navigation
Expand file tree
/
Copy pathGraph.java
More file actions
74 lines (64 loc) · 1.42 KB
/
Copy pathGraph.java
File metadata and controls
74 lines (64 loc) · 1.42 KB
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
public class Graph {
int count;
public Graph(int c){
this.count = c;
}
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
String line;
line = in.nextLine();
int T = Integer.parseInt(line);
while(T-->0){
solve(in);
}
}
private static void solve(Scanner in) {
String line;
String[] a;
line = in.nextLine();
int N = Integer.parseInt(line);
HashMap<String, Graph> map = new HashMap<String, Graph> ();
while(N-->0){
line = in.nextLine();
a = line.split("\\s+");
Graph g1 = map.get(a[0]);
Graph g2 = map.get(a[1]);
if(g1==null && g2==null){
g1 = new Graph(2);
map.put(a[0], g1);
map.put(a[1], g1);
}else if(g1==g2){
continue;
}else if(g1!=null && g2!=null){
merge(map, g1, g2);
}else if(g2==null && g1!=null){
map.put(a[1], g1);
g1.count ++;
}else if (g2!=null && g1==null){
map.put(a[0], g2);
g2.count ++;
}
System.out.println(map.get(a[0]).count);
}
}
private static void merge(HashMap<String, Graph> map, Graph g1, Graph g2) {
Graph tmp;
if(g1.count < g2.count){
tmp = g1;
g1 = g2;
g2 = tmp;
}
setValue(map,g2,g1);
}
private static void setValue(HashMap<String, Graph> map, Graph g2, Graph g1){
for(String k :map.keySet()){
if(map.get(k)==g2){
map.put(k, g1);
}
}
g1.count += g2.count;
}
}