Ter*_* Li 9 java sorting hashmap treemap comparator
我一直在寻找Map<String, Integer>按价值排序的方法.我发现这篇文章解决了我的排序问题,但并不完全正确.根据帖子,我写了以下代码:
import java.util.*;
public class Sort {
static class ValueComparator implements Comparator<String> {
Map<String, Integer> base;
ValueComparator(Map<String, Integer> base) {
this.base = base;
}
@Override
public int compare(String a, String b) {
if (base.get(a) >= base.get(b)) {
return 1;
} else {
return -1;
}
}
}
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<String, Integer>();
ValueComparator vc = new ValueComparator(map);
TreeMap<String, Integer> sorted = new TreeMap<String, Integer>(vc);
map.put("A", 1);
map.put("B", 2);
sorted.putAll(map);
for (String key : sorted.keySet()) {
System.out.println(key + " : " + sorted.get(key)); // why null values here?
}
System.out.println(sorted.values()); // But we do have non-null values here!
}
}
Run Code Online (Sandbox Code Playgroud)
输出:
A : null
B : null
[1, 2]
BUILD SUCCESSFUL (total time: 0 seconds)
Run Code Online (Sandbox Code Playgroud)
从输出中可以看出,该get方法始终返回null.原因是我的ValueComparator.compare()方法永远不会返回0,我已经通过发表这篇文章找到了.
有人在那篇文章中建议以下解决null价值问题:
public int compare(String a, String b) {
if (base.get(a) > base.get(b)) {
return 1;
}else if(base.get(a) == base.get(b)){
return 0;
}
return -1;
}
Run Code Online (Sandbox Code Playgroud)
我测试了这段代码,它引入了一个关键的合并问题.换句话说,当值相等时,它们的相应键被合并.
我也尝试过以下方法:
public int compare(String a, String b) {
if (a.equals(b)) return 0;
if (base.get(a) >= base.get(b)) {
return 1;
} else return -1;
}
Run Code Online (Sandbox Code Playgroud)
它也不起作用.有些价值观仍然存在null.此外,此解决方法可能存在逻辑问题.
有人可以为我的问题提出一个完全可行的解决方案 我希望按值排序功能和工作get方法同时工作.
在比较函数中,当值相等时,您应该比较键.这将确保具有相同值的不同密钥不会被"合并",因为它消除了否则将比较相等的条目.
例如:
@Override
public int compare(String a, String b) {
Integer x = base.get(a);
Integer y = base.get(b);
if (x.equals(y)) {
return a.compareTo(b);
}
return x.compareTo(y);
}
Run Code Online (Sandbox Code Playgroud)
(您需要修改上面的代码以匹配您的空值策略)
请注意,您对值进行排序的方法非常脆弱.您的"已排序"地图将不支持添加新条目,这可能会令人困惑.