栈最大的特点就是先入后出,先看一下简单的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 ();
}
数组实现的栈就完成了,它可以自动扩容,可以迭代。