In this post, I am going to explain how to implement custom Iterator and solve one interview question using iterator.
What is iterator?
Iterator is used to traverse the Collection of Elements. There is also a legacy class called Enumeration, it is also used to traverse collection of elements. Only difference is by using iterator you can remove an element while traversing, by using enumerator you can’t.
Let me brief the methods in iterator interface.
Method | Description |
boolean hasNext() | Returns true if the iteration has more elements. |
E next() | Returns the next element in the iteration. |
default void remove() | Removes from the underlying collection the last element returned by this iterator. Java provides default implementation for this method. The default implementation throws an instance of UnsupportedOperationException and performs no other action. |
Iterable interface
All the Java collections implementing Iterable interface, to return an iterator. Implementing this interface allows an object to be the target of the "foreach" statement. Following table summarizes the methods in Iterable interface.
Method | Description |
Iterator | Returns an iterator over a set of elements of type T. |
Now lets come to the problem, I want to implement my own iterator. Suppose my collection has elements like 1, 2, 3, 4, 5, 6, 7, 8, 9. My iterator should traverse the elements at even positions like 1(is at 0thposition), 3(is at 2nd position), 5, 7, 9.
I defined EvenList class like below.
import java.util.Iterator;
import java.util.List;
import java.util.NoSuchElementException;
public class EvenListT> implements IterableT> {
private ListT> list;
EvenList(ListT> list) {
this.list = list;
}
public IteratorT> iterator() {
return new EvenIteratorT>();
}
@SuppressWarnings("hiding")
private class EvenIteratorT> implements IteratorT> {
int size = list.size();
int currentPointer = 0;
public boolean hasNext() {
return (currentPointer size);
}
public T next() {
if (!hasNext()) {
throw new NoSuchElementException();
}
@SuppressWarnings("unchecked")
T val = (T) list.get(currentPointer);
currentPointer += 2;
return val;
}
}
}
Total logic is in the next() method, here I am checking for an element in the collection using hasNext() method, it it returns false, I am throwing NoSuchElementException(). If element exists, then I am saving current element to a temporary variable and incrementing the currentPointer to 2.
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
public class TestEvenList {
public static void main(String args[]) {
ListInteger> list = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
EvenListInteger> myList = new EvenList(list);
IteratorInteger> iter = myList.iterator();
while (iter.hasNext()) {
System.out.println(iter.next());
}
}
}
Run above program, you will get following output.
1
3
5
7
9
You may like
Difference between iterator and enumerator
Difference between Iterator and ListIterator
Why get method is synchronized in Hashtable?
Serialization Vs Singleton
Difference between HashMap and TreeMap with an Example
Difference between HashMap and TreeMap with an Example
This post first appeared on Java Tutorial : Blog To Learn Java Programming, please read the originial post: here
