Java收藏建议

Question

我有一个方法如下

  public List<List<CustomClass>> categorize(List<CustomClass> customClass){
        List<List<CustomClass>> returnValue = new ArrayList<>();
        for (CustomClass customClassValue: customClass) {
              List<CustomClass> one = new ArrayList<>(), two = new ArrayList<>(), three = new ArrayList<>(), four = new ArrayList<>();
              switch (customClassValue.getAge()){
              case 1:
                    one.add(customClassValue);
                    break;
              case 2:
                    two.add(customClassValue);
                    break;
              case 3:
                    three.add(customClassValue);
                    break;
              case 4:
                    four.add(customClassValue);
                    break;
              }
              returnValue.add(one);
              returnValue.add(two);
              returnValue.add(three);
              returnValue.add(four);
        }
        return returnValue;
  } 

从有效性和性能的角度来看，除了使用List<List<CustomClass>> 之外，java 集合中还有其他选择吗？

简单描述一下函数的作用：

获取自定义对象列表的输入，并根据对象中的字段对其进行分类，并分别返回每个类别项。

Answer 1

当您想将事物分类到桶中时，我的第一反应是使用Map 而不是List。根据年龄将List<CustomClass> 划分为子列表似乎是使用Map<Integer, List<CustomClass>> 的最佳时机。这是一种方法：

    public Map<Integer, List<CustomClass>> categorize(List<CustomClass> customClass) {
        Map<Integer, List<CustomClass>> returnValue = new HashMap<>();

        for (CustomClass customClassValue: customClass) {
            List<CustomClass> sublist = returnValue.get(customClassValue.getAge());

            if (sublist == null) {
                sublist = new ArrayList<>();
                returnValue.put(customClassValue.getAge(), sublist);
            }

            sublist.add(customClassValue);
        }

        return returnValue;
    }

Answer 2

Multimap 是Map<Key, List<Value>> 的更简洁版本，Google 的 Guava 有一个很好的实现。它应该可以解决您遇到的问题。

Answer 3

试试这个。

public List<List<CustomClass>> categorize(List<CustomClass> customClass) {
    return customClass.stream()
        .filter(v -> v.getAge() >= 1 && v.getAge() <= 4)
        .collect(
            () -> IntStream.range(0, 4)
                .mapToObj(i -> new ArrayList<CustomClass>())
                .collect(Collectors.toList()),
            (list, v) -> list.get(v.getAge() - 1).add(v),
            (a, b) -> {});
}