下压堆栈的两种实现

算法第四版学习笔记一（下压堆栈的实现）

使用数组和链表两种实现方式
数组实现方式：

package ren.laughing.algorithms;

import java.util.Iterator;

/**
 * 下压堆栈（数组实现）
 * 
 * @author Laughing_Lz
 * @createTime 2018年9月2日
 * @param <E>
 */
public class ResizingArrayStack<E> implements Iterable<E> {

  private E[] items = (E[]) new Object[1];
  private int N = 0;

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

  public int size() {
    return N;
  }

  /**
   * 动态调整数组大小 保持数组大小与栈大小之比小于一个常数
   * 
   * @param max 调整后的数组长度
   */
  public void resize(int max) {
    E[] temp = (E[]) new Object[max];
    for (int i = 0; i < N; i++) {
      temp[i] = items[i];
    }
    items = temp;
  }

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

  public E pop() {
    E item = items[--N];
    // 避免对象游离
    items[N] = null;
    if (N > 0 && N == items.length / 4) {
      resize(items.length / 2);
    }
    return item;
  }

  /**
   * 实现foreach遍历
   */
  @Override
  public Iterator<E> iterator() {
    return new StackIterator();
  }

  private class StackIterator implements Iterator<E> {
    private int i = N;

    @Override
    public boolean hasNext() {
      return i > 0;
    }

    @Override
    public E next() {
      return items[--i];
    }

  }
}

复制代码

链表实现方式：

package ren.laughing.algorithms;

import java.util.Iterator;

/**
 * 下压堆栈（链表实现）
 * 
 * @author Laughing_Lz
 * @createTime 2018年9月2日
 * @param <E>
 */
public class Stack<E> implements Iterable<E> {

  private Node first;
  private int N;

  private class Node {
    E item;
    Node next;
  }

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

  public int size() {
    return N;
  }

  public void push(E item) {
    Node oldfirst = first;
    first = new Node();
    first.item = item;
    first.next = oldfirst;
    N++;
  }

  public E pop() {
    E item = first.item;
    first = first.next;
    N--;
    return item;
  }

  /**
   * 实现foreach遍历
   */
  @Override
  public Iterator<E> iterator() {
    return new StackIterator();
  }

  private class StackIterator implements Iterator<E> {

    private Node current = first;

    @Override
    public boolean hasNext() {
      return first != null;
    }

    @Override
    public E next() {
      E item = current.item;
      current = first.next;
      return item;
    }

  }
}

复制代码