自定义实现迭代器——高淇JAVA300讲笔记之迭代器

        这段代码分别以不同的方式实现了iterator()方法，实现了add()方法在末尾添加元素。自定义的Iterator对象具有hasNext()、next()、remove()方法。数组的拷贝用到了Arrays.copyOf方法，这个方法传回的数组是一个新的数组对象。

package com.bjsxt.iter;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;

/**
 * 目标：深入迭代器的原理  --> 一个容器可以创建多个迭代器对象
 * 途径：使用了 内部类 |匿名内部类
 * 
 * 深入：
 * 1.使用内部类 实现迭代器
 * 2.使用Iterable 实现foreach迭代
 * 3.加入末尾添加元素的方法
 * 
 *
 */
public class SxtDeepList implements java.lang.Iterable{
    //数组 存储值
    private String[] elem = new String[5];
    //实际大小
    private int size = 0;
    
    public int size() {
        return this.size;
    }
    
    /**
     * 在末尾添加元素
     * @param ele
     */
    public void add(String ele) {
        if(this.size == elem.length) {  //容量不够 --> 扩容
            elem = Arrays.copyOf(elem, elem.length+5);
            
        }
        
        elem[size] = ele;  //数组中加入元素 加到最后
        size++;//实际大小+1
    }
    
    //内部类
    private class MyIter implements Iterator {
        //计数器 就是指针 游标
        private int cursor = -1;
        //判断是否存在下一个
        public boolean hasNext() {
            return cursor+1 < size;
        }
        //获取下一个
        public String next() {
            cursor++;
            return elem[cursor];
        }
        //删除下一个
        public void remove() {
            //移动数组元素
            System.arraycopy(elem, 
                    cursor+1, elem, cursor, 
                    size-(cursor+1));
            //实际大小-1
            size--;
            //回退
            this.cursor--;
        }
    }
    
    public Iterator iterator1() {
        return new MyIter();
    }
    
    public Iterator iterator2() {
        class MyIter implements Iterator {
            //计数器 就是指针 游标
            private int cursor = -1;
            //判断是否存在下一个
            public boolean hasNext() {
                return cursor+1 < size;
            }
            //获取下一个
            public String next() {
                cursor++;
                return elem[cursor];
            }
            //删除下一个
            public void remove() {
                //移动数组元素
                System.arraycopy(elem, 
                        cursor+1, elem, cursor, 
                        size-(cursor+1));
                //实际大小-1
                size--;
                //回退
                this.cursor--;
            }
        }
        return new MyIter();
    }
    
    public Iterator iterator() {
        return new Iterator() {//创建Iterator迭代器接口实现类(没有名称)的对象
            //计数器 就是指针 游标
            private int cursor = -1;
            //判断是否存在下一个
            public boolean hasNext() {
                return cursor+1 < size;
            }
            //获取下一个
            public String next() {
                cursor++;
                return elem[cursor];
            }
            //删除下一个
            public void remove() {
                //移动数组元素
                System.arraycopy(elem, 
                        cursor+1, elem, cursor, 
                        size-(cursor+1));
                //实际大小-1
                size--;
                //回退
                this.cursor--;
            }
        };
    }
    
    public static void main(String[] args) {
        SxtDeepList list = new SxtDeepList();
        list.add("a");
        list.add("b");
        list.add("c1");
        list.add("c2");
        list.add("c3");
        list.add("c4");
        list.add("c5");
        Iterator it = list.iterator();
        while(it.hasNext()) {  //先判断 后获取
            System.out.println(it.next());
            it.remove();
        }
        System.out.println(list.size());
        
        for(Object str:list) {
            System.out.println(str);
        }
        
        
        ArrayList list2 = new ArrayList();
        list2.add("a");
        list2.add("a");
        list2.add("a");
        for(Object obj:list2) {  //foreach
            System.out.println(obj);
        }
    }
}