forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution.java
43 lines (43 loc) · 1.27 KB
/
Solution.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
class Solution {
public int maximumDetonation(int[][] bombs) {
int n = bombs.length;
List<Integer>[] g = new List[n];
Arrays.setAll(g, k -> new ArrayList<>());
for (int i = 0; i < n - 1; ++i) {
for (int j = i + 1; j < n; ++j) {
int[] p1 = bombs[i], p2 = bombs[j];
double dist = Math.hypot(p1[0] - p2[0], p1[1] - p2[1]);
if (dist <= p1[2]) {
g[i].add(j);
}
if (dist <= p2[2]) {
g[j].add(i);
}
}
}
int ans = 0;
boolean[] vis = new boolean[n];
for (int k = 0; k < n; ++k) {
Arrays.fill(vis, false);
vis[k] = true;
int cnt = 0;
Deque<Integer> q = new ArrayDeque<>();
q.offer(k);
while (!q.isEmpty()) {
int i = q.poll();
++cnt;
for (int j : g[i]) {
if (!vis[j]) {
vis[j] = true;
q.offer(j);
}
}
}
if (cnt == n) {
return n;
}
ans = Math.max(ans, cnt);
}
return ans;
}
}