Mutual Followers
import java.util.*;
class Solution {
public int[] solve(int[][] relations) {
Set<Pair<Integer, Integer>> g = new HashSet<>();
for (int[] relation: relations) {
g.add(new Pair(relation[0], relation[1]));
}
Set<Integer> s = new HashSet<>();
for (Pair<Integer, Integer> edge: g) {
int other = edge.getValue();
int person = edge.getKey();
if (g.contains(new Pair(other, person))) {
s.add(person);
s.add(person);
}
}
int[] ret = new int[s.size()];
int i = 0;
for (int n: s) {
ret[i] = n;
i++;
}
Arrays.sort(ret);
return ret;
}
}
Last updated