2 回答
TA贡献1802条经验 获得超5个赞
您的方向是正确的,但您需要考虑第一次找到邮政编码/产品组合时会发生什么。
在 Java 的更高版本中,有很多Map方法可以使这变得更容易。我将在这里使用它们,但如果您必须使用早期版本,那么您将需要扩展其中一些语句。
像下面这样:
Map<String, Map<Product, Integer>> zipCodeProducts = new HashMap<>();
for (Customer customer: customers) {
Map<Product,Integer> productCounts = zipCodeProducts.computeIfAbsent(customer.getZipCode(), () -> new HashMap<>());
for (Purchase purchase: customer.getItems()) {
productCounts.merge(purchase.getProduct(), 1, Integer::sum);
}
}
获得计数最高的产品应该相对简单:
Map<String,Integer> maxProducts = new HashMap<>();
zipCodeProducts.forEach((zc, pc) -> pc.forEach((pr, n) -> {
if (!maxProducts.contains(zc) || n > pc.get(maxProducts.get(zc)))
maxProducts.put(zc, pr);
}));
希望这是有道理的——如果没有的话就问。
TA贡献1804条经验 获得超7个赞
我认为您希望将 if 语句移至 for 循环的开头,并且仅tmpMap在该邮政编码尚不存在时才创建。如果它已经存在,只需使用现有的并使用产品和数量更新它。
for (Customer customer : this.customers) {
String tmp = customer.getZipCode();
Map<Product, Integer> tmpMap;
if (!zipCodeProducts.containsKey(tmp)){
tmpMap = new HashMap<Product, Integer>();
} else {
tmpMap = zipCodeProducts.get(tmp);
}
for (Purchase purchase: customer.getItems()) {
if (!tmpMap.containsKey(purchase.getProduct())) {
tmpMap.put(purchase.getProduct(),purchase.getAmount());
} else {
tmpMap.put(purchase.getProduct(), tmpMap.get(purchase.getProduct()) + purchase.getAmount());
}
}
zipCodeProducts.put(tmp, tmpMap);
}
添加回答
举报