我需要創建一個函式 public Collection findEpidemics(List diagnostics, int k, int n) 來確定疾病是否是流行病，并回傳流行病的串列。如果一種疾病在連續 k 天內發生超過 n 次，我們就說它是一種流行病。Diagnosis 類是 public Diagnosis(Disease disease, int day) 并且具有函式 getDisease() 和 getDay()。
串列診斷如下所示：
Disease cholera = new Disease("cholera");
Disease dengue = new Disease("dengue");
List<Diagnosis> diagnoses = Arrays.asList(
new Diagnosis(cholera, 0), // registered cholera on day 0
new Diagnosis(cholera, 0),
new Diagnosis(cholera, 1),
new Diagnosis(dengue, 2),
new Diagnosis(dengue, 2),
// we have a gap here
new Diagnosis(cholera, 6),
所以我所做的是制作一個哈希圖收集，其中包含每天發生的所有疾病，因此它可能看起來像這樣：{{cholera, 0}=2, {cholera, 1}=1, {dengue , 2}=2, {霍亂, 6}=1}
public Collection<String> findEpidemics(List<Diagnosis> diagnoses, int k, int n) {
// first we count occurrences
Map<DiagnosisMetric, Long> collect = diagnoses.stream().
collect(Collectors.groupingBy(DiagnosisMetric::new, counting()));
// check for epidemic
for(int i=0; i<collect.size(); i ){
for(int j=0; j<k; j ){
???
}
}
}
我有點堅持檢查疾病是否是流行病，所以我堅持如何回圈遍歷哈希圖并檢查連續 k 天某種疾病的發生率是否超過 n。
uj5u.com熱心網友回復：
獲得映射后，您可以使用條目串列上的滑動視窗繼續檢查流行病。
public static void findEpidemics(List<Diagnosis> diagnoses, int k, int n) {
Map<DiagnosisMetric, Long> collect = diagnoses.stream().collect(Collectors.groupingBy(DiagnosisMetric::new, counting()));
Map<String, List<Map.Entry<DiagnosisMetric, Long>>> diseaseByName = collect.entrySet().stream().collect(Collectors.groupingBy(diagnosisMetricLongEntry -> diagnosisMetricLongEntry.getKey().getDiagnosis().getDisease().getName()));
List<String> pandemics = diseaseByName.entrySet().stream().collect(Collectors.partitioningBy(e -> isPandemic(e, n, k))).getOrDefault(true, Collections.emptyList()).stream().map(m -> m.getKey()).collect(Collectors.toList());
System.out.println(pandemics);
}
public static boolean isPandemic(Map.Entry<String, List<Map.Entry<DiagnosisMetric, Long>>> e, int n, int k) {
List<Map.Entry<DiagnosisMetric, Long>> collect = e.getValue().stream().sorted(Comparator.comparingInt(o -> o.getKey().getDiagnosis().getDay())).collect(Collectors.toList());
int s = collect.size();
Long sum = 0L;
if(s < k) {
return false;
}
Queue<Long> q = new ArrayDeque<>();
for(int i = 0; i < k; i ) {
Long value = collect.get(i).getValue();
q.add(value);
sum = value;
if(sum > n) {
return true;
}
}
for(int j = k; j < s; j ) {
Long value = collect.get(j).getValue();
Long lastValue = q.poll();
sum = sum - lastValue;
q.add(value);
sum = value;
if(sum > n) {
return true;
}
}
return false;
}
實作是相當基本的，但您可以為此使用滑動視窗技術。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/321838.html