从 TreeSet 中删除元素时出现问题

Question

我正在执行以下操作

class RuleObject implements Comparable{



    @Override
    public String toString() {
        return "RuleObject [colIndex=" + colIndex + ", probability="
                + probability + ", rowIndex=" + rowIndex + ", rule=" + rule
                + "]";
    }
    String rule;
    double probability;
    int rowIndex;

    int colIndex;

    public RuleObject(String rule, double probability) {
        this.rule = rule;
        this.probability = probability;
    }
    @Override
    public int compareTo(Object o) {

        RuleObject ruleObj = (RuleObject)o;
        System.out.println(ruleObj);
        System.out.println("---------------");
        System.out.println(this);
        if(ruleObj.probability > probability)
            return 1;
        else if(ruleObj.probability < probability)
            return -1;
        else{
            if(ruleObj.colIndex == this.colIndex && ruleObj.rowIndex == this.rowIndex && ruleObj.probability == this.probability && ruleObj.rule.equals(this.rule))
                return 0;
        }
        return 1;

    }


}

并且我有一个包含 RuleObject 元素的 TreeSet。 我正在尝试执行以下操作：

System.out.println(sortedHeap.size());
        RuleObject ruleObj = sortedHeap.first();
        sortedHeap.remove(ruleObj);
System.out.println(sortedHeap.size());

我可以看到集合的大小保持不变。我无法理解为什么它不被删除。 另外，在删除时我可以看到调用了compareTo 方法。但它只被调用了 3 个对象，而 set 中有 8 个对象。 谢谢

Answer 1

查看 删除：

从该集合中删除指定元素（如果存在）。更正式地说，如果该集合包含这样的元素，则删除 e 使得 (o==null ? e==null : o.equals(e)) 。

您的问题是 RuleObject 不 @Override equals(Object other)。您需要这样做，当然，您还需要@Override hashCode()。

---

此外，compareTo 被调用的次数少于元素数量的原因是因为它应该是一个 O(log N) 操作；这就是使用 TreeSet 的全部目的。如果您有 1024 个元素，则 compareTo 的调用次数预计不会超过 10 次。

---

正如弗拉德指出的那样，你的比较器坏了。具体来说，最后一个语句 return 1; 破坏了它。您应该根据其他字段扩展相等的概率情况以返回 -1、0、+1。

Answer 2

正如polygenelubricants 所指出的，您必须在RuleObject 上实现equals。

此外，您的 比较器 本质上是破碎的。它不会强加 总排序，即在某些情况下，它会声明 RuleObject a 既大于又小于另一个 RuleObject b （例如，如果 a.probability= =b.probability 和 a.colIndex != b.colIndex。）这将导致在树插入、遍历等过程中出现不需要的行为

。最后，compareTo 还必须与equals一致，即

C 类的自然排序是
表示与 equals 一致 if
并且仅当 (e1.compareTo((Object)e2)
== 0) 对于每个 e1 和 e1.equals((Object)e2) 具有相同的布尔值
C 类的 e2。注意 null 不是
任何类的实例，以及
e.compareTo(null) 应该抛出一个
NullPointerException 即使
e.equals(null) 返回 false。

如果您不关心 RuleObject 的顺序，请使用 HashSet。

否则（即您想要以明确定义的方式迭代 TreeSet，例如优先级、顺序），请实现 Comparable 以考虑所有感兴趣的字段，例如：

public int compareTo(Object o) {
  RuleObject r = (RuleObject)o;
  // assume no nulls for now, but you should eventually check
  // also assume o is always of type RuleObject for now,
  //  but you should eventually check
  return
    priority < r.priority ? -1 :
    priority > r.priority ? 1 :
    colIndex < r.colIndex ? -1 :
    colIndex > r.colIndex ? 1 :
    rowIndex < r.rowIndex ? -1 :
    rowIndex > r.rowIndex ? 1 : 0;
}

boolean equals(Object o) {
  // Delegate to compareTo(); no code duplication, consistent.
  return compareTo(o) == 0;
}

Answer 3

如果您希望它正常工作™，请尝试 Apache 的 CompareToBuilder，HashCodeBuilder 和 EqualsBuilder。我提供了示例代码，可以从下面开始。您当然不必采取这条路线，但我认为您会发现它简化了事情。请注意不同函数之间的一致模式。

public int compareTo(Object o) {
  RuleObject ruleObj = (RuleObject) o;
  return new CompareToBuilder()
    .append(this.probability, ruleObj.probability)
    .append(this.colIndex, ruleObj.colIndex)
    .append(this.rowIndex, ruleObj.rowIndex)
    .append(this.rule, ruleObj.rule)
    .toComparison();
}

public int hashCode() {
  // You should customize the hard-coded numbers, as described in the docs.
  return new HashCodeBuilder(17, 37).
    append(probability).
    append(colIndex).
    append(rowIndex).
    append(rule).
    toHashCode();
}

public boolean equals(Object obj) {
  if (obj == null) { return false; }
  if (obj == this) { return true; }
  if (obj.getClass() != getClass()) {
    return false;
  }
  RuleObject rhs = (RuleObject) obj;
  return new EqualsBuilder()
    .append(probability, rhs.probability)
    .append(colIndex, rhs.colIndex)
    .append(colIndex, rhs.colIndex)
    .append(rule, rhs.rule)
    .isEquals();
}