更新時間:2023-08-01 來源:黑馬程序員 瀏覽量:
在Java中,我們可以自己編寫一個容器類,并且可以使用for-each循環(huán)(也稱為增強型for循環(huán)或foreach循環(huán))遍歷容器中的元素。要實現這一點,我們的容器類必須滿足以下條件:
我們的容器類必須實現java.lang.Iterable接口。該接口包含一個名為iterator() 的方法,該方法返回一個實現了java.util.Iterator接口的對象,用于遍歷容器中的元素。
我們的容器類還必須實現java.util.Iterator接口。Iterator接口定義了幾個方法,如hasNext()(判斷是否有下一個元素)、next()(返回下一個元素)和remove()(從容器中移除當前元素,可選操作)。
接下來筆者用一個具體的代碼示例,演示下如何創(chuàng)建一個自定義容器類并使用for-each循環(huán)遍歷它:
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 循環(huán)遍歷容器 for (String fruit : container) { System.out.println(fruit); } } }
輸出結果:
Apple Banana Orange
在上述例子中,我們創(chuàng)建了一個自定義容器類MyContainer,實現了Iterable接口,并提供了一個內部類 MyIterator實現了Iterator接口。這樣,我們就可以使用for-each循環(huán)來遍歷容器中的元素。