使TreeMap Comparator容忍null

时间:2012-03-07 06:57:05

标签: java sorting nullpointerexception comparator treemap

这个自定义的Valuecomparator按其值对TreeMap进行排序。但是在搜索TreeMap是否具有某个键时,它不会容忍nullpointexception。如何修改比较器以处理空值?

    import java.io.IOException;
    import java.util.Comparator;
    import java.util.HashMap;
    import java.util.Map;
    import java.util.TreeMap;



    public class TestTreeMap {

        public static class ValueComparator<T> implements Comparator<Object> {

            Map<T, Double> base;
            public ValueComparator(Map<T, Double> base) {
                this.base = base;
            }

            @Override
            public int compare(Object a, Object b) {
                /*if (((Double) base.get(a) == null) || ((Double) base.get(b) == null)){
                    return -1;
                }   */      
                if ((Double) base.get(a) < (Double) base.get(b)) {
                    return 1;
                } else if ((Double) base.get(a) == (Double) base.get(b)) {
                    return 0;
                } else {
                    return -1;
                }
            }

        }

        public static void main(String[] args) throws IOException { 
            Map<String, Double> tm = new HashMap<String, Double>();
            tm.put("John Doe", new Double(3434.34)); 
            tm.put("Tom Smith", new Double(123.22)); 
            tm.put("Jane Baker", new Double(1378.00)); 
            tm.put("Todd Hall", new Double(99.22)); 
            tm.put("Ralph Smith", new Double(-19.08)); 

            ValueComparator<String> vc = new ValueComparator<String>(tm);
            TreeMap<String, Double> sortedTm = 
                    new TreeMap<String, Double>(vc);
            sortedTm.putAll(tm);

            System.out.println(sortedTm.keySet());
            System.out.println(sortedTm.containsKey("John Doe"));
            // The comparator doesn't tolerate null!!!
            System.out.println(!sortedTm.containsKey("Doe"));
        }


}

1 个答案:

答案 0 :(得分:6)

这不是火箭科学......

将其插入已注释掉的代码:

if (a == null) {
    return b == null ? 0 : -1;
} else if (b == null) {
    return 1;
} else 

这会将null视为比任何非null Double实例更小的值。


您的版本不正确:

if ((a==null) || (b==null)) {return -1;}

这表示“如果a为null或b为null,则a小于b”。

这会导致像

这样的虚假关系
null < 1.0  AND 1.0 < null

null < null

当set / map中存在空值时,这种事情导致树不变量中断,并导致不一致和不稳定的键排序......更糟糕。

有效compare方法的要求javadocs中列出。数学版本是该方法必须在所有可能输入值的域上定义total order