当前位置: 首页 > 面试题库 >

如何根据地图的“集合”值的大小对地图进行排序?

何楷
2023-03-14
问题内容

我有一个HashMap这样的:

Map<String, List<String>> map = new HashMap<>();

map.put("USA", Arrays.asList("CA","IA","IL"));
map.put("India", Arrays.asList("MUM","CAL"));
map.put("Canada", Arrays.asList("TOR"));

我想根据列表值的大小按升序对地图进行排序。我怎样才能做到这一点?

在这种情况下,我希望订购加拿大,印度,美国的钥匙。


问题答案:

HashMap没有可保证的迭代顺序,因此您需要收集到一个LinkedHashMap才能使排序有意义。

import static java.util.Comparator.comparingInt;
import static java.util.stream.Collectors.toMap;

Map<String, List<String>> sorted = map.entrySet().stream()
    .sorted(comparingInt(e -> e.getValue().size()))
    .collect(toMap(
        Map.Entry::getKey,
        Map.Entry::getValue,
        (a, b) -> { throw new AssertionError(); },
        LinkedHashMap::new
    ));

AssertionError之所以引发,是因为合并器功能仅用于并行流],而我们并未使用。

comparingByValue如果您觉得可读性更好,也可以使用:

import static java.util.Map.Entry.comparingByValue;

Map<String, List<String>> sorted = map.entrySet().stream()
    .sorted(comparingByValue(comparingInt(List::size)))
    // ... as above


 类似资料: