在Java 8中,您的整个代码可以像这样编写(如提出的其他精细答案):
Map> myHashMap = new HashMap();
for (Person person : persons) {
myHashMap.computeIfAbsent(age,age->new ArrayList()).add(person);
}
但是通过使用收集到带有Collectors.GroupingBy()的Map的流可以更短.
Map> myMap = persons.stream().collect(Collectors.groupingBy(Person:getAge));
作为旁注,您的实际Java 7代码也可以得到改进.
当然,没有Java 8那么多,但如果你不能使用Java 8,这可能会很有趣.
在您的实际代码中,这是重复的:
personsOfSameAge.add(person);
并且您使用两个条件语句(if和else),而只有在您首先处理特殊情况时才足够:在Map中没有值.
这是一个修改版本:
Map> myHashMap = new HashMap<>();
for (Person person : persons) {
int age = person.getAge();
List personsOfSameAge = myHashMap.get(age);
if (personsOfSameAge == null) {
personsOfSameAge = new ArrayList();
myHashMap.put(age, personsOfSameAge);
}
personsOfSameAge.add(person);
}