我正在嘗試將對象從添加List<Farm>到Map<Animal, List<Farm>>public class Farm { private String farmName; private EnumSet<Animal> animals = EnumSet.noneOf(Animal.class); /* ... */}Farm f1 = new Farm("Farm A", EnumSet.of(Animal.CAT, Animal.DOG, Animal.DUCK));Farm f2 = new Farm("Farm B", EnumSet.of(Animal.PIG, Animal.CAT, Animal.HORSE));Farm f3 = new Farm("Farm C", EnumSet.of(Animal.DUCK));任務 1：將對象添加到List<Farm>List<Farm> list = new ArrayList<>();list.add(f1);list.add(f2);list.add(f3);任務 2：將列表中的對象添加到地圖（鍵：Animal，值：）List <Farm>我以這種方式完成了這項任務：Map<Animal, List<Farm>> map = new HashMap<>();for(Farm farm: list) { for(Animal an: farm.getAnimals()) { if(!map.containsKey(an)) { List<Farm> new_list = new ArrayList<>(); new_list.add(farm); map.put(an, new_list); }else { List<Farm> old_list = map.get(an); if(!old_list.contains(farm)) { old_list.add(farm); } } } }有第二個/更有效的解決方案嗎？是這樣的：Map<Animal, List<Farm>> map = list.stream().collect(Collectors.groupingBy(Farm::getAnimals)));這不起作用，因為getAnimals返回一個EnumSet<Animal>.
2 回答
茅侃侃
TA貢獻1842條經驗 獲得超21個贊
您可能希望保留循環，但對其進行現代化改造：
Map<Animal, List<Farm>> map = new EnumMap<>(Animal.class);
for(Farm farm: list)
for(Animal an: farm.getAnimals())
map.computeIfAbsent(an, x -> new ArrayList<>()).add(farm);
在你的循環中，add(farm)冗余出現在兩個分支中，因為你總是將它添加到List. 然后，computeIfAbsent允許消除條件，因為它將返回一個現有值或構造一個新值，放入并返回它。收集groupingBy器在內部也使用這種方法。
使用 Stream 操作的缺點是您需要臨時持有兩個值，例如
Map<Animal, List<Farm>> map = list.stream()
.flatMap(farm -> farm.getAnimals().stream()
.map(animal -> new AbstractMap.SimpleImmutableEntry<>(animal, farm)))
.collect(Collectors.groupingBy(Map.Entry::getKey,
() -> new EnumMap<>(Animal.class),
Collectors.mapping(Map.Entry::getValue, Collectors.toList())));
溫溫醬
TA貢獻1752條經驗 獲得超4個贊
我從對面開始，但我想這很有幫助
Map<Animal, List<Farm>> map = Arrays.stream(Animal.values()) .collect(Collectors.toMap(an -> an, an -> list.stream().filter(f -> f.getAnimals().contains(an)).collect(Collectors.toList())));
在沒有任何農場包含動物的情況下，動物可能有空集，但這可以很容易地過濾
添加回答
舉報
0/150
提交
取消