-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path2493.java
36 lines (36 loc) · 1.14 KB
/
2493.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
class Solution {
public int magnificentSets(int n, int[][] edges) {
List<Integer>[] g = new List[n];
Arrays.setAll(g, k -> new ArrayList<>());
for (var e : edges) {
int a = e[0] - 1, b = e[1] - 1;
g[a].add(b);
g[b].add(a);
}
int[] d = new int[n];
int[] dist = new int[n];
for (int i = 0; i < n; ++i) {
Deque<Integer> q = new ArrayDeque<>();
q.offer(i);
Arrays.fill(dist, 0);
dist[i] = 1;
int mx = 1;
int root = i;
while (!q.isEmpty()) {
int a = q.poll();
root = Math.min(root, a);
for (int b : g[a]) {
if (dist[b] == 0) {
dist[b] = dist[a] + 1;
mx = Math.max(mx, dist[b]);
q.offer(b);
} else if (Math.abs(dist[b] - dist[a]) != 1) {
return -1;
}
}
}
d[root] = Math.max(d[root], mx);
}
return Arrays.stream(d).sum();
}
}