数组实现简单的栈结构

栈最大的特点就是先入后出，先看一下简单的API
public int size() //返回栈中数据个数
public void push(Item item) //进栈
public Item pop() //出栈
public boolean isEmpty() //是否为空
自定义的栈要实现迭代器功能实现 Iterable接口，需要存储所有的结构所以使用泛型。

具体代码实现:

import java.util.Iterator;
//使用泛型来适合所有的数据类型
public class MyStack<Item> implements Iterable<Item>{
    private Item[] a;
    private int N;
    //初始化数组
    public MyStack(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(a.length*2);
        }
        a[N++] = item;
    }
    //出栈
    public Item pop(){
        Item item = a[--N];
        //把出栈的那个空间设为null,java垃圾回收机制会回收那个空间，不赋为null不会回收它
        a[N] = null;
        //如果数组的长度太长则需减小长度
        if(N<a.length/4){
            resize(a.length/2);
        }
    }
    //扩容,缩容,使用private来只让本类调用
    privete void resize(int cap){
        Item[] temp = (Item[]) new Object[cap];
        for(int i=0;i<a.length;i++){
            temp[i] = a[i];
        }
        a = temp;
    }
    //返回迭代器对象
    @Override
    public Iterator<Item> iterator() {
        //此处代码下面讲解
    }
}

基本的栈操作已经实现了，接下来实现迭代功能，迭代就是可以用象for(Item item:items)来遍历栈,实现迭代需要返回一个迭代器对象，迭代器对象要实现Iterator接口。代码如下:

//在MyStarc类中定义一个 MyStarcIterator类实现Iterator接口,里面有两个要实现的方法
    private class MyStackIterator implements Iterator<Item> {
        //获取外部类中的N，用来迭代
        private int i = N;
        //hasNext()来判断栈中是否还有数据
        @Override
        public boolean hasNext() {
            return i>0;
        }
        //用来获取遍历的对象，先入后出
        @Override
        public Item next(){
            return a[--i];
        }
    }

上面还有一个空方法
@Override
public Iterator iterator()
这个方法返回一个我们写的迭代器对象，即：

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

数组实现的栈就完成了，它可以自动扩容，可以迭代。

数组实现简单的栈结构

猜你喜欢