当我们自己编写一个集合类,并希望能够使用增强型 for 循环(for-each loop)来遍历集合内的元素,需要满足以下两个条件:
-
实现 Iterable 接口:集合类必须实现 Iterable 接口,该接口定义了一个方法 iterator(),该方法返回一个实现了 Iterator 接口的对象。通过实现 Iterable 接口,我们向外部提供了一种获取迭代器的能力,从而使得我们的集合类可以被 for 循环遍历。
-
实现迭代器(Iterator):在集合类内部,需要定义一个实现了 Iterator 接口的迭代器类。迭代器类需要实现 hasNext() 和 next() 方法,用于判断是否还有下一个元素以及获取下一个元素。迭代器类可以是集合类的内部类,也可以是独立的类。
当满足上述条件后,我们就可以使用增强型 for 循环来遍历集合内的元素。在 for 循环中,编译器会自动调用集合类的 iterator() 方法获取迭代器对象,然后使用迭代器对象进行迭代。
以下是一个简单示例,演示了自定义集合类 MyCollection 的使用:
import java.util.Iterator;
public class Main {
public static void main(String[] args) {
MyCollection<String> collection = new MyCollection<>();
collection.add("Apple");
collection.add("Orange");
collection.add("Pear");
for (String s : collection) {
System.out.println(s);
}
}
}
class MyCollection<T> implements Iterable<T> {
private T[] elements;
private int size;
@SuppressWarnings("unchecked")
public MyCollection() {
elements = (T[]) new Object[10];
size = 0;
}
public void add(T element) {
elements[size] = element;
size++;
}
@Override
public Iterator<T> iterator() {
return new MyIterator();
}
private class MyIterator implements Iterator<T> {
private int index;
public MyIterator() {
index = 0;
}
@Override
public boolean hasNext() {
return index < size;
}
@Override
public T next() {
T element = elements[index];
index++;
return element;
}
}
}
在上面的示例中,我们定义了一个自定义集合类 MyCollection,它实现了 Iterable 接口并提供了一个内部类 MyIterator 实现迭代器。
我们可以使用 add() 方法向 MyCollection 中添加元素,并通过实现 Iterable 接口和返回 MyIterator 的 iterator() 方法,使得我们可以使用增强型 for 循环来遍历集合内的元素。
通过实现迭代器的 hasNext() 和 next() 方法,我们可以在 MyIterator 类中定义迭代器的行为,使得可以按照一定的顺序迭代集合内的元素。在 MyIterator 的 hasNext() 方法中,判断当前索引是否小于集合的大小,以确定是否还有下一个元素可以迭代。在 next() 方法中,我们返回当前索引处的元素,并将索引递增,以准备获取下一个元素。
这样,通过实现 Iterable 接口和迭代器,我们的自定义集合类 MyCollection 就可以使用增强型 for 循环来遍历集合内的元素了。在 main 方法中,我们创建了一个 MyCollection 对象 collection,添加了几个字符串元素,然后使用增强型 for 循环遍历并打印出每个元素。
通过自定义集合类和迭代器,我们可以根据自己的需求实现特定的迭代逻辑,使集合的使用更加灵活和符合业务需求。