import java.util.HashMap;
import java.util.HashSet;
class Solution {
public int[] solution(String[] id_list, String[] report, int k) {
HashMap<String, String> userMap = new HashMap<String, String>();
HashMap<String, HashSet<String>> stopMap = new HashMap<>();
for (int i = 0; i < report.length; i++) {
String[] reportSplit = report[i].split(" ");
HashSet<String> userSet = new HashSet<String>();
if(stopMap.containsKey(reportSplit[1])) {
userSet = stopMap.get(reportSplit[1]);
}
userSet.add(reportSplit[0]);
stopMap.put(reportSplit[1], userSet);
}
int[] result = new int[id_list.length];
for(String i : stopMap.keySet()) {
if(stopMap.get(i).size() >= k) {
for (int j = 0; j < id_list.length; j++) {
if(stopMap.get(i).contains(id_list[j])) {
result[j]++;
}
}
}
}
return result;
}
}
import java.util.HashMap;
import java.util.HashSet;
class Solution {
public int[] solution(String[] id_list, String[] report, int k) {
HashMap<String, HashSet<String>> stopMap = new HashMap<>();
HashSet<String> userSet = new HashSet<String>();
for (int i = 0; i < report.length; i++) {
String[] reportSplit = report[i].split(" ");
if(stopMap.containsKey(reportSplit[1])) {
userSet = stopMap.get(reportSplit[1]);
}
userSet.add(reportSplit[0]);
stopMap.put(reportSplit[1], userSet);
}
int[] result = new int[id_list.length];
for(String i : stopMap.keySet()) {
if(stopMap.get(i).size() >= k) {
for (int j = 0; j < id_list.length; j++) {
if(stopMap.get(i).contains(id_list[j])) {
result[j]++;
}
}
}
}
return result;
}
}