[英]Add to List value on HashMap
我正在填充一个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);
是的,Java 8中有:
List<Person> personsOfSameAge = myHashMap.computeIfAbsent(age,age->new ArrayList<Person>());
personsOfSameAge.add(person);
要不就
myHashMap.computeIfAbsent(age,age->new ArrayList<Person>()).add(person);
在Java 8中,您的整个代码可以像这样编写(如提出的其他精细答案):
Map<Integer, List<Person>> myHashMap = new HashMap();
for (Person person : persons) {
myHashMap.computeIfAbsent(age,age->new ArrayList<Person>()).add(person);
}
但是通过使用收集到带有Collectors.groupingBy()
的Map
的流可以更短。
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);
}
像这样的东西?
List<Person> newList = Optional.ofNullable( myHashMap.get(age)).orElse(new ArrayList<>());
newList.add(person);
myHashMap.put(age, newList);
虽然问题涉及在Java 8中进行,但我将其作为替代方式发布。
您可以使用Google Guava中的MultiMap 。
Multimap<String, Person> myHashMap = ArrayListMultimap.create();
for (Person person : persons) {
myHashMap.put(person.getAge(), person);
}
但是,您必须意识到不同之处:Java的地图为不存在的键返回null
,而Google Guava的MultiMap返回empty List
。
从javadoc 获取 MultiMap的get(key)
返回与此多图中的键关联的值的视图集合(如果有)。 请注意,当containsKey(key)为false时,返回空集合,而不是null。 对返回的集合的更改将更新基础多图,反之亦然
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.