java - 树状图对象的get方法返回空值

Question

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!
    }
}

输出：

A : null
B : null
[1, 2]
BUILD SUCCESSFUL (total time: 0 seconds)

我想知道为什么我们在第一个注释行得到空值，而我们确实有非空值，如第二个注释行所示。

编辑：@null 的版本似乎不起作用。我已将代码更改如下：

        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;
        }

它似乎有效，但我不确定。

Answer 1

我的猜测是您的ValueComparator.compare()方法永远不会返回 0，表示相等，导致该Map.get()方法找不到匹配项。

Answer 2

以这种方式更改您的比较

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;  
    }

Answer 3

即使你的比较器肯定坏了，如果你把它改成

for (Map.Entry e : sorted.entrySet()) {
    System.out.println(e.getKey() + " : " + e.getValue());
}