使用 ENUMMAP 替代序数索引

Posted 2020-11-25 wcss

import java.util.Arrays;
import java.util.EnumMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

public class Herb {
public enum Type {
ANNUAL, PERENNTAL, BIENNIAL
}

private final String name;
private final Type type;

Herb(String name, Type type) {
this.name = name;
this.type = type;
}

@Override
public String toString() {
return name;
}

public static void main(String[] args) {
// 将集合放到一个按照类型的序数进行索引的数组中来实现 替换
Herb[] garden = { new Herb("一年生1", Type.ANNUAL), new Herb("一年生2", Type.ANNUAL), new Herb("两年生1", Type.BIENNIAL),
new Herb("两年生2", Type.BIENNIAL), new Herb("多年生1", Type.PERENNTAL), new Herb("多年生2", Type.PERENNTAL) };

//    @SuppressWarnings("unchecked")//产生泛型数组，且手工操作索引
//    Set<Herb>[] herbsByType = (Set<Herb>[]) new Set[Herb.Type.values().length];
//    for (int i = 0; i < herbsByType.length; i++) {
//    herbsByType[i] = new HashSet<Herb>();
//    }
//    for (Herb h : garden) {
//    herbsByType[h.type.ordinal()].add(h);
//    //System.out.println(h.type);
//    }
//    for (int i = 0; i < herbsByType.length; i++) {
//    System.out.printf("%s: %s%n", Herb.Type.values(), herbsByType[i]);
//    }

Map<Herb.Type, Set<Herb>> herbsByType = new EnumMap<Herb.Type, Set<Herb>>(Herb.Type.class);
for (Herb.Type t : Herb.Type.values()) {
herbsByType.put(t, new HashSet<Herb>());
}
for (Herb h : garden) {
herbsByType.get(h.type).add(h);
}
System.out.println(herbsByType);



//如果采用的是stream
long startTime = System.currentTimeMillis(); 
// 这段代码的问题在于，它选择了自己的 map 实现，而实际上它不是 EnumMap，所以它不会将版本的空间和时间性能与显式 EnumMap 匹配。
System.out.println(Arrays.stream(garden).collect(Collectors.groupingBy(p -> p.type)));
long endTime = System.currentTimeMillis();
System.out.println("程序运行时间：" + (endTime - startTime) + "ms"); 
结果：{PERENNTAL=[多年生1, 多年生2], ANNUAL=[一年生1, 一年生2], BIENNIAL=[两年生1, 两年生2]}
程序运行时间：45ms

// 要纠正这个问题，可以使用 Collectors.groupingBy 的三参数形式，它允许调用者使用 mapFactory 参数指定 Map 实现：
long startTime2 = System.currentTimeMillis(); 
System.out.println(Arrays.stream(garden)
.collect(Collectors.groupingBy(p -> p.type, () -> new EnumMap<>(Type.class), Collectors.toSet())));
long endTime2 = System.currentTimeMillis();
System.out.println("程序运行时间：" + (endTime2 - startTime2) + "ms"); 
}
}

结果： {ANNUAL=[一年生2, 一年生1], PERENNTAL=[多年生1, 多年生2], BIENNIAL=[两年生2, 两年生1]}

程序运行时间：1ms