在HashMap上添加到列表值

Question

我正在填充一个hashmap,让我的对象按其属性之一进行分组.我发现在将对象添加到列表之前检查列表(值)是否存在"丑陋".

一个例子将更明确:

// Need a map to group Person by age.
// Person = {age: int}
Map<Integer, List<Person>> myHashMap = new HashMap();

for (Person person : persons) {
   int age = person.getAge();
   List<Person> personsOfSameAge = myHashMap.get(age);
   if (personsOfSameAge != null) {
       personsOfSameAge.add(person);
   } else {
       personsOfSameAge = new ArrayList();
       personsOfSameAge.add(person);
       myHashMap.put(age, personsOfSameAge);
   }
}

有没有更好的方法来编码？

myHashMap.addToValueListOrCreateValueList(myObject);

Answer 1

在Java 8中,您的整个代码可以像这样编写(如提出的其他精细答案):

Map<Integer, List<Person>> myHashMap = new HashMap();

for (Person person : persons) {
   myHashMap.computeIfAbsent(age,age->new ArrayList<Person>()).add(person);
}

但是,你可能会更通过收集成一个流短Map带Collectors.groupingBy():

Map<Integer, List<Person>> myMap = persons.stream().collect(Collectors.groupingBy(Person:getAge));

作为旁注,您的实际Java 7代码也可以得到改进.当然,没有Java 8那么多,但如果你不能使用Java 8,这可能会很有趣.
在您的实际代码中,这是重复的:

personsOfSameAge.add(person);

并且您使用两个条件语句(if和else),而只有if在您首先处理特殊情况时才足够:没有值Map.
这是一个修改版本:

Map<Integer, List<Person>> myHashMap = new HashMap<>();

for (Person person : persons) {
   int age = person.getAge();
   List<Person> personsOfSameAge = myHashMap.get(age);
   if (personsOfSameAge == null) {
       personsOfSameAge = new ArrayList();
       myHashMap.put(age, personsOfSameAge);
   }          
   personsOfSameAge.add(person);                
}

Answer 2

是的,Java 8中有:

List<Person> personsOfSameAge = myHashMap.computeIfAbsent(age,age->new ArrayList<Person>());
personsOfSameAge.add(person);

要不就

myHashMap.computeIfAbsent(age,age->new ArrayList<Person>()).add(person);