java - 如何重写 Set 的删除方法？

Question

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

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 内的声明.