最近项目用到了treeMap做排序并分类计数,当order字段相同时出现分组混乱:
代码:
public static Map<String, List<KtIndicatorResult>> getresultTreeMap() {
Map<String, List<KtIndicatorResult>> resultMap = new TreeMap<>(new Comparator<String>() {
@Override
public int compare(String o1, String o2) {
// 截取key:INDICATOR_GROUP + '_' + GROUP_ORDER里面的GROUP_ORDER
Integer com1 = Integer.valueOf(o1.substring(o1.lastIndexOf(SEPARATER2) + 1,o1.lastIndexOf(KtUtils.SEPARATER1)));
Integer com2 = Integer.valueOf(o2.substring(o2.lastIndexOf(SEPARATER2) + 1,o2.lastIndexOf(KtUtils.SEPARATER1)));
return com1.compareTo(com2);
}
});
return resultMap;
}
问题:
当com1.compareTo(com2) 返回为0 时,分类出现问题,会把com1和com2对应的key划分到同一类中去:
解决:
查看TreeMap源码发现:
1 : comparator.compare(k1, k2) 返回值小于0 ,则把值放在左子树,相当于排在前面.
2 : comparator.compare(k1, k2) 返回值大于0 ,则把值放在右子树,相当于排在后面.
3 : comparator.compare(k1, k2) 返回值等于0 ,把k1和k2当成相同key,并替换值。
最后把基础数据中每个分组对应的order值设为不同,解决问题,后期再考虑其他方式.
参考: