Question

我试图为Set编写remove方法，但在测试时它不起作用。我做错了吗？删除元素后，大小不会减小。

public class MySet<T> implements Set<T> {

  private T[] arrayB;
  private int elements;

  @Override
  public boolean remove(Object f) {
      T h = (T) f;
      for (T z : arrayB) {
          if(z == h) {
              z = null;
          }
      }
      return true;
  }

  @Override
  public int size() {
      int count = 0;
      for(int arr = 0; arr < arrayB.length; arr++){
          if(arrayB[arr] != null) {
              count++;
          }
      }
      return count;
  }

测试代码是：

MySet<Integer> ints = new MySet<Integer>();
    for (int i = 0; i < 100; i++) {
        ints.add(i);
    }
    for (int i = 0; i < 100; i += 2) {
        ints.remove(i);
    }   
}

Answer 1

您的size方法依赖于元素是否null来决定是否计算它。假设您试图在数组中放置null，那么您做错了。您所做的就是将null分配给z，这只是一个局部变量。数组没有改变。

您必须使用传统的for循环并使用数组访问表达式将null分配给数组元素。您还需要致电equals而不是使用==来查找元素。

for (int i = 0; i < array.length; i++)
{
    if (array[i] != null && array[i].equals(h))
    {
        array[i] = null;
    }
}

根据您是要删除匹配的所有元素，还是仅删除第一个元素，您可以考虑在break中添加if语句。

如何覆盖Set的remove方法？

1 个答案: