SortedSet‹TestClass› сравнение на равенство по одному полю и сортировка по другому

Пожалуйста, просмотрите код:

/* Run1.java */
package test;

import java.util.Iterator;
import java.util.SortedSet;
import java.util.TreeSet;

public class Run1 
{
    static public void main(String[] args)
    {
        SortedSet<TestClass> s = new TreeSet<TestClass>(); 

        s.add( new TestClass("name1", 100) );
        s.add( new TestClass("name2", 10) );
        s.add( new TestClass("name3", 1) );
        s.add( new TestClass("name4", 10) );
        s.add( new TestClass("name5", 100) );

        Iterator<TestClass> it = s.iterator();

        while(it.hasNext())
        {
            TestClass t = it.next();
            System.out.println( t.name+' '+t.value );
        }
    }
}

/* TestClass.java */
package test;

public class TestClass implements Comparable<TestClass> 
{
    public String name;
    public int value;

    public TestClass(String name, int value) {
        this.name = name;
        this.value = value;
    }

    public int compareTo(TestClass o) 
    {
        return this.value - o.value;
    }

    public boolean equals(Object o) 
    {
        if (!(o instanceof TestClass))
            return false;
        TestClass n = (TestClass)o;

        return this.name.equals(n.name);
    }

    public int hashCode() 
    {
        return 31*name.hashCode();
    }

    public String toString() 
    {
        return name;
    }
}

Распечатка

name3 1
name2 10
name1 100

как я вижу, потому что compareTo используется для проверки на равенство (когда возвращается 0). Но мне нужно проверить уникальность по полю TestClass.name и отсортировать только по TestClass.value


java sortedset treeset
person triclosan    schedule 16.11.2011    source источник
comment
Ваш вопрос не имеет смысла. TreeSet использует метод compareTo(), а не методы equals() или hashCode(), для определения порядка сортировки (что является обобщением равенства в контексте отсортированного множества). Используйте другую структуру данных, если это не является желаемой семантикой. Тем не менее: предостережение emptor - есть очень веские причины, по которым набор деревьев не использует equals() для равенства объектов. Что, если a.compareTo(b) — это 0, а a.equals(b) — это false? Что, если a.compareTo(b) не равно нулю, а a.equals(b) равно true?   -  person Matt Ball    schedule 16.11.2011
comment
спасибо! Посоветуйте подходящий контейнер данных. Мне нужно быстро получить элементы по ключу (TestClass.value), но ключ может дублироваться   -  person triclosan    schedule 16.11.2011

Ответы (4)


arrow_upward
2
arrow_downward

В этом случае результат compareTo() и equals() должен быть совместимым, а это значит, что при сравнении необходимо учитывать равенство. Например:

public int compareTo(TestClass o) 
{
    return (this.value == o.value) ? this.name.compareTo(o.name) : this.value - o.value;
}

который вводит подпорядок имени для объектов с одинаковым значением, что делает результат совместимым с вашей equals() реализацией.

person rsp    schedule 16.11.2011

arrow_upward
2
arrow_downward

как насчет взлома метода compareTo следующим образом:

public int compareTo(TestClass o) 
{
    if (this.name != null && this.name.equals(o.name)) {
        return 0;
    }


    return this.value - o.value;
}

Это должно выполнять проверку на равенство по имени (удалять дубликаты) при сортировке по значению.

person aishwarya    schedule 16.11.2011

arrow_upward
1
arrow_downward

Если я правильно понимаю, то вы хотите, чтобы ваш compareTo всегда реализовывал «естественный порядок» для класса. Это означает, что клиент класса ожидает от класса поведения. Согласно контракту, compareTo должен быть совместим с equals, поэтому я всегда реализую equals следующим образом:

return compareTo(obj)==0;

Это гарантирует согласованность.

Затем, если вам нужен другой порядок сортировки, вы должны реализовать другой класс, реализующий Comparable. Таким образом, вы можете иметь согласованность классов и отдельные порядки сортировки.

person Thom    schedule 16.11.2011

arrow_upward
0
arrow_downward

Напишите компаратор, который сравнивает объекты TestClass.

public class TVComparator implements Comparator<TestClass> {
    public int compare(TestClass o1, TestClass o2) {
        if (o1.name.equals(o2.name)) return 0;
        return o1.value - o2.value;
    }
}

Ради простоты я опустил любые проверки нулевых значений.

person RokL    schedule 16.11.2011