【Java】第二十四节 详述ArrayList类remove方法和contains方法

remove&contains

• 以ArrayList为例，如果其中存的是基本数据类型的包装类或是String类，则remove()方法和contains()方法很容易实现。
• 例：以String类为例：

  public class Test {
  	public static void main(String[] args) {
  		ArrayList<String> list = new ArrayList<String>();
  		list.add("Jing");
  		list.add("Peng");
  		list.add("Chong");
  		
  		System.out.println(list.size());	//3
  		System.out.println(list.contains("Jing"));	//true
  		list.remove("Jing");
  		System.out.println(list.size());	//2
  	}
  }	
  

• 但是如果ArrayList中存储的是自定义类，结果将会不同
• 创建有名对象与将有名对象添加到集合中分开执行这种方式与基本数据类型的处理类似，以自定义类TestClass为例：

  class TestClass {
  	String id;
  	public TestClass(String id) {
  		this.id = id;
  	}
  }
  public class RemoveDemo {
  	public static void main(String[] args) {
  		ArrayList<TestClass> list = new ArrayList<>();
  		TestClass tc1 = new TestClass("001");
  		TestClass tc1 = new TestClass("002");
  		TestClass tc1 = new TestClass("003");
  		list.add(tc1);
  		list.add(tc2);
  		list.add(tc3);
  		System.out.println(list.contains(tc1));	//true
  		list.remove(tc1);
  		System.out.println(list.size());	//2
  	}
  }
  

• 但是创建匿名对象与将匿名对象添加到集合中一起执行这种方式，结果就不同了，例：

  class TestClass {
  	String id;
  	public TestClass(String id) {
  		this.id = id;
  	}
  }
  public class RemoveDemo {
  	public static void main(String[] args) {
  		ArrayList<TestClass> list = new ArrayList<>();
  		list.add(new TestClass("001"));
  		list.add(new TestClass("002"));
  		list.add(new TestClass("003"));
  		
  		System.out.println(list.contains(new TestClass("002")));	//false
  		list.remove(new TestClass("002"));
  		System.out.println(list.size());	//3
  	}
  }
  

• remove()方法的实现原理是先判断传进来的对象是否为null，如果为null则在集合中查找null，找到了就将其删除并返回true，没找到就返回false；如果不为null，则在集合中调用对象的equals()方法(由于此时没有重写equals()方法，所以调用了父类Object的equals()方法)逐个判断有没有与该对象的地址相同的元素，如果有就将其删除并返回true，没有就返回false，其源码如下：

  //该方法是ArrayList的方法：
  public boolean remove(Object o) {
      if (o == null) {
          for (int index = 0; index < size; index++)
              if (elementData[index] == null) {
                  fastRemove(index);
                  return true;
              }
      } else {
          for (int index = 0; index < size; index++)
              if (o.equals(elementData[index])) {
                  fastRemove(index);
                  return true;
              }
      }
      return false;
  }
  //上面调用的equals()是Object的方法，该方法如下：
  public boolean equals(Object obj) {
      return (this == obj);
  }
  

• contains()方法的实现原理是返回indexOf(o) >= 0的值，indexOf()先判断传进来的对象是否为null，如果为null则在集合中查找null，找到了就返回所在下标，没找到就返回-1；如果不为null，则在集合中调用对象的equals()方法(由于此时没有重写equals()方法，所以调用了父类Object的equals()方法)逐个判断有没有与该对象的地址相同的元素，如果有就返回其下标，没有就返回-1，其源码如下：

  public boolean contains(Object o) {
          return indexOf(o) >= 0;
  }
  public int indexOf(Object o) {
      if (o == null) {
          for (int i = 0; i < size; i++)
              if (elementData[i]==null)
                  return i;
      } else {
          for (int i = 0; i < size; i++)
              if (o.equals(elementData[i]))
                  return i;
      }
      return -1;
  }
  

• 引用数据类型存储的是地址，而每new一个对象，都会在堆里面新开辟一块空间，因此尽管有对象的id相同，但它们并不是同一个对象，每个对象的地址是不同的。如果想要以id是否相同来判断是否为一个对象，那么就需要重写equals()方法。String字符串集合可以与基本数据类型包装类集合一样直接使用remove()和contains()就是因为其重写了父类object的equals()方法。例：

  class TestClass {
  	String id;
  	public TestClass(String id) {
  		this.id = id;
  	}
  	@Override
  	public boolean equals(Object obj) {
  		if(obj instanceof TestClass) {
  			TestClass tc = (TestClass)obj;
  			return this.id.equals(tc.id);
  		}
  		return false;
  	}
  }
  public class RemoveDemo {
  	public static void main(String[] args) {
  		ArrayList<TestClass> list = new ArrayList<>();
  		list.add(new TestClass("001"));
  		list.add(new TestClass("002"));
  		list.add(new TestClass("003"));
  		System.out.println(list.contains(new TestClass("002")));	//true
  		list.remove(new TestClass("002"));
  		System.out.println(list.size());	//2
  	}
  }