Сортировка HashMap в TreeMap: пользовательский компаратор удаляет значения с тем же ключом - PullRequest
Новая
       15

Сортировка HashMap в TreeMap: пользовательский компаратор удаляет значения с тем же ключом

0 голосов
/ 07 октября 2018

Я пытаюсь отсортировать HashMap<String, Long>.У меня есть следующий код для сортировки: 

private static class ValueComparator implements Comparator<String>{
            HashMap<String, Long> map = new HashMap<String, Long>();

            public ValueComparator(HashMap<String, Long> map){
                this.map.putAll(map);
            }

            @Override
            public int compare(String s1, String s2) {
                if(map.get(s1) > map.get(s2)){
                    System.out.println("s1: " + s1 + "; s2: " + s2);
                    return -1;
                }
                else if (map.get(s1).equals(map.get(s2))) {
                    return 0;
                }
                else{
                    return 1;
                }
            }
        }

private static TreeMap<String, Long> sortMapByValue(HashMap<String, Long> map){
                Comparator<String> comparator = new ValueComparator(map);
                //TreeMap is a map sorted by its keys.
                //The comparator is used to sort the TreeMap by keys.
                TreeMap<String, Long> result = new TreeMap<String, Long>(comparator);
                result.putAll(map);

                System.out.println("DONE sort");
                return result;
        }

Проблема в том, что, когда несколько разных ключей имеют одинаковые значения, только один ключ превращает его в окончательную карту:

ПРИМЕР: 

 public class Test  {
    public static void main(String[] args)  {
        HashMap<String, Long> hashMap = new HashMap<>();
        hashMap.put("Cat", (long) 4);
        hashMap.put("Human", (long) 2);
        hashMap.put("Dog", (long) 4);
        hashMap.put("Fish", (long) 0);
        hashMap.put("Tree", (long) 1);
        hashMap.put("Three-legged-human", (long) 3);
        hashMap.put("Monkey", (long) 2);

        System.out.println(hashMap);  //7 pairs

        System.out.println(sortMapByValue(hashMap));  //5 pairs
   }
}

Как бы это исправить?

1 Ответ

0 голосов
/ 07 октября 2018

Я не думаю, что это исправимо, вы используете карты непреднамеренно и нарушаете контракты.Предполагается, что древовидная карта будет отсортирована по ключу, а ключ должен быть уникальным, поэтому при сравнении == 0 он просто переопределит значение узла.Вы всегда можете реализовать свой собственный TreeMap и заставить его делать то, что вы хотите.

Я не уверен, что вы хотите с ним делать, но я думаю, что вам нужно что-то вроде 

TreeMap<Long,List<String>>

http://hg.openjdk.java.net/jdk8/jdk8/jdk/file/687fd7c7986d/src/share/classes/java/util/TreeMap.java

    if (cpr != null) {
        do {
            parent = t;
            cmp = cpr.compare(key, t.key);
            if (cmp < 0)
                t = t.left;
            else if (cmp > 0)
                t = t.right;
            else
                return t.setValue(value);
        } while (t != null);
...