更新时间:2023-08-01 来源:黑马程序员 浏览量:
在Java中,我们可以自己编写一个容器类,并且可以使用for-each循环(也称为增强型for循环或foreach循环)遍历容器中的元素。要实现这一点,我们的容器类必须满足以下条件:
我们的容器类必须实现java.lang.Iterable接口。该接口包含一个名为iterator() 的方法,该方法返回一个实现了java.util.Iterator接口的对象,用于遍历容器中的元素。
我们的容器类还必须实现java.util.Iterator接口。Iterator接口定义了几个方法,如hasNext()(判断是否有下一个元素)、next()(返回下一个元素)和remove()(从容器中移除当前元素,可选操作)。
接下来笔者用一个具体的代码示例,演示下如何创建一个自定义容器类并使用for-each循环遍历它:
import java.util.Iterator; // 自定义容器类 class MyContainer<T> implements Iterable<T> { private T[] elements; private int size; @SuppressWarnings("unchecked") public MyContainer(int capacity) { elements = (T[]) new Object[capacity]; size = 0; } public void add(T element) { elements[size++] = element; } @Override public Iterator<T> iterator() { return new MyIterator(); } // 自定义迭代器类 private class MyIterator implements Iterator<T> { private int currentIndex = 0; @Override public boolean hasNext() { return currentIndex < size; } @Override public T next() { return elements[currentIndex++]; } } } public class Main { public static void main(String[] args) { MyContainer<String> container = new MyContainer<>(3); container.add("Apple"); container.add("Banana"); container.add("Orange"); // 使用 for-each 循环遍历容器 for (String fruit : container) { System.out.println(fruit); } } }
输出结果:
Apple Banana Orange
在上述例子中,我们创建了一个自定义容器类MyContainer,实现了Iterable接口,并提供了一个内部类 MyIterator实现了Iterator接口。这样,我们就可以使用for-each循环来遍历容器中的元素。