编写函数findepidemic

Question

我需要创建一个函数public Collection findedemic (List diagnostics, int k, int n)，用于确定某种疾病是否为流行病，并返回流行病的疾病列表。如果一种疾病在连续k天内发生n次以上，我们就称其为流行病。Diagnosis类是公共的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),

我所做的是创建一个hashmap集合，其中包含每天所有疾病的发生情况例如它可能看起来像这样:{{霍乱，0}=2，{霍乱，1}=1，{登革热，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

Answer 1

获得映射后，您可以使用条目列表上的滑动窗口继续检查大流行。

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;
}

实现是相当初级的，但是您可以使用滑动窗口技术来实现。