Question

我意识到有类似的问题，但他们没有回答我的问题。

我需要根据相应值的ArrayList的大小返回HashMap的键。例如，如果我有：

HashMap<String,ArrayList<Integer>> unsortedMap = new HashMap<String,ArrayList<Integer>>();
unsortedMap.put("A",new ArrayList<Integer>(Arrays.asList(1,2,3)));
unsortedMap.put("B",new ArrayList<Integer>(Arrays.asList(4)));
unsortedMap.put("C",new ArrayList<Integer>(Arrays.asList(2,3,1,4)));

我希望它返回“C”“A”“B”

Answer 1

对于Java 7，您可以致电entrySet()获取Set<Map.Entry<String,ArrayList<Integer>>> - 然后您可以使用它来填充ArrayList<Map.Entry<String,ArrayList<Integer>>>之类的内容，您可以使用自定义比较器对其进行排序。

import java.util.*;

public class Test {
  public static void main(String[] args) {
    Map<String, ArrayList<Integer>> unsortedMap = new HashMap<>();
    unsortedMap.put("A", new ArrayList<Integer>(Arrays.asList(1, 2, 3)));
    unsortedMap.put("B", new ArrayList<Integer>(Arrays.asList(4)));
    unsortedMap.put("C", new ArrayList<Integer>(Arrays.asList(2, 3, 1, 4)));

    List<Map.Entry<String, ArrayList<Integer>>> list = 
        new ArrayList<>(unsortedMap.entrySet());
    Collections.sort(list, new EntryComparator());

    for (Map.Entry<String, ArrayList<Integer>> entry : list) {
      System.out.println(entry.getKey());
    }
  }

  private static class EntryComparator
      implements Comparator<Map.Entry<String, ArrayList<Integer>>>
  {
    public int compare(Map.Entry<String, ArrayList<Integer>> left,
        Map.Entry<String, ArrayList<Integer>> right) {     
      // Right then left to get a descending order
      return Integer.compare(right.getValue().size(), left.getValue().size());
    }
  }
}

在Java 8中，您可以使用流API使其更加流畅 - 同时采取基本相同的步骤。

import java.util.*;
import java.util.stream.*;

public class Test {
  public static void main(String[] args) {
    Map<String, ArrayList<Integer>> unsortedMap = new HashMap<>();
    unsortedMap.put("A", new ArrayList<Integer>(Arrays.asList(1, 2, 3)));
    unsortedMap.put("B", new ArrayList<Integer>(Arrays.asList(4)));
    unsortedMap.put("C", new ArrayList<Integer>(Arrays.asList(2, 3, 1, 4)));

    List<String> keys = unsortedMap
          .entrySet()
          .stream()
          .sorted((left, right) ->
              Integer.compare(right.getValue().size(), left.getValue().size()))
          .map(entry -> entry.getKey())
          .collect(Collectors.toList());

    for (String key : keys) {
      System.out.println(key);
    }
  }
}

Answer 2

如果您使用java-8，也可以这样做：

List<String> keys = unsortedMap.entrySet()
                   .stream()
                   .sorted((e1, e2) -> Integer.compare(e2.getValue().size(), e1.getValue().size()))
                   .map(Map.Entry::getKey)
                   .collect(Collectors.toList());
System.out.println(keys); //[C, A, B]

它的作用是：

获取地图条目的流
按照每个arraylist的大小对条目进行排序
将每个条目映射到其对应的密钥
在List

如果您愿意，您也可以将排序的行写为：

.sorted(Comparator.comparing(e -> e.getValue().size(), Comparator.reverseOrder()))

基于ArrayList5 Size对ArrayList的HashMap进行排序

2 个答案: