集合数据类型(自实现栈)实现迭代

先上代码

package fzu.com;

import java.util.Iterator;

public class FixedCapacityStack<Item> implements Iterable<Item> {
    private  Item[] a;
    private int N;

    public FixedCapacityStack(int cap) {
        a = (Item[]) new Object[cap];
    }

    public boolean isEmpty() {
        return N == 0;
    }

    public int size() {
        return N;
    }

    public void push(Item item) {
        if(N == a.length) resize(2 * a.length);
        a[N++] = item;
    }

    public Item pop() {
        Item item = a[--N];
        a[N] = null;
        if(N > 0 && N == a.length / 4) resize(a.length / 2);
        return item;
    }

    private void resize(int max)  {
        Item[] temp = (Item[]) new Object[max];
        for(int i = 0; i < N; i++){
            temp[i] = a[i];
        }
        //if (N >= 0) System.arraycopy(a, 0, temp, 0, N);另一种写法
        a = temp;
    }

    @Override
    public Iterator<Item> iterator() {
        return new ReverseArrayIterator();
    }

    private class ReverseArrayIterator implements Iterator<Item>{

        private int i = N;
        @Override
        public boolean hasNext() {
            return i > 0;
        }

        @Override
        public Item next() {
            return a[--i];
        }

        @Override
        public void remove() {

        }
    }

    public static void main(String[] args) {
        FixedCapacityStack<Integer> testStack = new FixedCapacityStack<Integer>(10);
        testStack.push(1);
        testStack.push(2);
        testStack.push(3);
        for (Integer c: testStack) {
            System.out.println(c);
        }
    }
}

  • 实现接口Iterable中的iterator方法,该方法返回一个迭代器Iterator
  • 还需实现Iterator中的hasNext()next()两个方法
  • 最后方可使用foreach进行迭代
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容