13

我在 TreeSet 中添加了一些简单的对象,但是当我调用 TreeSet 的 remove() 和 contains() 方法时,它们不起作用。但是,当我遍历集合时,会打印对象。员工对象应添加到集合中,而对象的唯一性基于对象名称属性。Id 属性是应该排序的值,但它不是唯一的。

public class Employee {
    private String name;
    private int id;

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
    return name;
    }

    public void setName(String name) {
    this.name = name;
    }

 // Two objects are considered equal if their names are equal
    @Override
    public boolean equals(Object o) {
    if (o == null)
        return false;
    if (this == o)
        return true; 
    if (o.getClass() == this.getClass()) {
        Employee p = ( Employee) o;
        if (p.getName() != null && this.getName() != null)
        return this.getName().equals(p.getName());
        else
        return false;
    } else {
        return false;
    }
    }
} 

//*******************************************************

public class EmployeeComp implements Comparator<Employee> {

    // Sort Ids, but allow duplicates, hence this function is never returning 0
    @Override
    public int compare(Employee p1, Employee p2) {
    int re = 0;

    boolean scoreLt = (p1.getId() > p2.getId());
    boolean scoreGt = (p1.getId() < p2.getId());

    if(scoreLt)
        re = -1;
    if(scoreGt)
        re = 1;
    else 
        re = -1;                       
         return re;                 
    }    
}
//*******************************************************
// Collection shall store unique names with ordered values like:
// Alex, 923
// Toni, 728
// Eddi, 232
// Peter, 232
// Eddi, 156  *** not allowed
import java.util.TreeSet;


public class Main {
    private static EmployeeComp comp = new EmployeeComp(); 
    private static TreeSet<Employee> employees = new TreeSet<Employee>(comp); 

    public static void main(String[] args) {

    Employee p1 = new Employee();
    p1.setName("Eddi");
    p1.setId(232);

    Employee p2 = new Employee();
    p2.setName("Toni");
    p2.setId(728);

    Employee p3 = new Employee();
    p3.setName("Peter");
    p3.setId(232);

    Employee p4 = new Employee();
    p4.setName("Alex");
    p4.setId(923);

    employees.add(p1);
    employees.add(p2);
    employees.add(p3);
    employees.add(p4);

    // Here, contains() and remove() should check the object address
    // and not perform their actions based on compareTo

       } 
}
4

2 回答 2

32

A根据 的结果TreeSet插入/删除,而不是/ Comparable.equals().hashCode()

这意味着,顺便说一句,您的对象Set确实实现了Comparable(如果它们没有实现,那么每次您尝试并插入一个成员时,您都会受到一个 的欢迎ClassCastException)。

更准确地说,TreeSetSortedSet.

如果你想要一个.equals()/.hashCode()兼容的集合,例如,使用一个HashSet.

为了说明,这是发生的事情BigDecimal(几个小时前在这里发布):

final BigDecimal one = new BigDecimal("1");
final BigDecimal oneDotZero = new BigDecimal("1.0");

final Set<BigDecimal> hashSet = new HashSet<>();
// BigDecimal implements Comparable of itself, so we can use that
final Set<BigDecimal> treeSet = new TreeSet<>();

hashSet.add(one);
hashSet.add(oneDotZero);
// hashSet's size is 2: one.equals(oneDotZero) == false

treeSet.add(one);
treeSet.add(oneDotZero);
// treeSet's size is... 1! one.compareTo(oneDotZero) == 0

引用 javadoc 的Comparable意思是BigDecimal's.compareTo()是“不一致的.equals()”。

** 编辑 ** 至于 OP 想要什么:

  • aCollection不接受重名;
  • Collection将根据用户 ID 进行排序的排序视图。

如上所述,你不能有一个集合同时做这两个。解决方案:

  • 首先,一个HashSet
  • 第二,将该集合的副本放入ArrayList,然后使用Collections.sort()

这意味着.equals()并且.hashCode()必须只对名称起作用,而习惯Comparator将对 id 起作用。没有其他选择,只能定制,因为它是一个在任何情况下Comparator都不符合的比较器。.equals()

至于建议的代码,有问题。

第一:Employee覆盖.equals()但不是.hashCode(). 因此,Employee违反了.equals()合同(其中一部分是如果两个对象相等,则它们必须具有相同的哈希码)。更重要的是,对于工作.hashCode()来说至关重要。HashSet使固定:

@Override
public int hashCode()
{
    return name == null ? 0 : name.hashCode();
}

@Override
public boolean equals(final Object obj)
{
    if (obj == null)
        return false;
    if (this == obj)
        return false;
    if (!(obj instanceof Employee))
        return false;
    final Employee other = (Employee) obj;
    return name == null ? other.name == null
        : name.equals(other.name);
}

第二:比较器同样被破坏,Employee因为它破坏了Comparator合同(对于任何o1o2o1.compareTo(o2) == - o2.compareTo(o1))。使固定:

public final class EmployeeComp
    implements Comparator<Employee>
{
    @Override
    public int compare(final Employee o1, final Employee o2)
    {
        final int id1 = o1.getId(), id2 = o2.getId();
        if (id1 == id2)
            return 0;
        return id1 > id2 ? 1 : -1;
    }
}

然后,如何获取集合的排序副本:

// "set" contains the unique employees
final List<Employee> sorted = new ArrayList<Employee>(set);
Collections.sort(list, new EmployeeComp());

完毕。

于 2013-06-06T19:16:08.627 回答
3

你的问题是概念性的。

如果你想要一个唯一对象的排序集合:TreeSet
如果你想要一个排序集合,不同的对象可以有相同的比较值用于排序目的:PriorityQueue

顺便说一句,PriorityList 中的方法比 TreeSet 中的方法更适合第二种情况的通常需要。我曾经认为它是 TreeSet 的缺点。例如,从集合中取出第一个项目。

希望有帮助:-)

于 2013-09-29T21:43:46.003 回答