TreeMap

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()方法找不到匹配项。