[Data Structure] Java 스택(Stack) 사용

• 이 글은 "온라인 자바 스터디 내용"을 정리한 글입니다.

Stack이란?

• 스택은 책을 쌓는 것처럼 데이터를 담을 수 있는 자료구조 입니다.

• 스택의 가장 큰 특징은 First In Last Out (FILO) 구조를 가지고 있습니다.

배열로 스택 구현

public class Stack {

    final int MAX_SIZE = 100;
    int arr[];
    int size;

    public Stack(){
        arr = new int[MAX_SIZE];
        size = 0;
    }

    public void push(int data){
        if(size<MAX_SIZE){
            arr[size++] = data;
        }
    }

    public int pop(){
        if(size>0){
            size--;
            return 1;
        }
        return 0;
    }

    public int top(){
        if(size>0){
            return arr[size-1];
        }
        return -1;
    }

    public boolean isFull(){
        if(size == MAX_SIZE){
            return true;
        }
        return false;
    }
}

public class Main {
    public static void main(String[] args) {
        Stack stack = new Stack();
        stack.push(1);
        stack.push(2);
        stack.push(3);

        System.out.println(stack.top());        // 3
        stack.pop();
        System.out.println(stack.top());        // 2
        stack.pop();
        System.out.println(stack.top());        // 1
        stack.pop();
    }
}

노드로 스택 구현

public class ListNode {
    int data;
    ListNode next;

    public ListNode(int data) {
        this.data = data;
        next = null;
    }
}

다음 클래스는 노드를 구현하는 코드입니다.

public class ListNodeStack {

    ListNode head;
    int size;

    public ListNodeStack() {
        head = null;
        size = 0;
    }

    void push(int data){
        ListNode node = head;
        if(size == 0){
            node = new ListNode(data);
            head = node;
        }
        else{
            while(node.next != null){
                node = node.next;
            }
            node.next = new ListNode(data);
        }
        size++;
    }

    int pop(){
        ListNode node = head;
        int data;
        if(size == 1){
            data = node.data;
            node = null;
            head = node;
        }
        else{

            while(node.next.next!=null){
                node = node.next;
            }
            data = node.next.data;
            node.next = null;
        }
        size--;
        return data;
    }

    public int getSize() {
        return size;
    }
}

public class Main {
    public static void main(String[] args) {

        ListNodeStack stack = new ListNodeStack();
        stack.push(1);
        stack.push(2);
        stack.push(3);

        System.out.println(stack.getSize());        // 3

        System.out.println(stack.pop());        // 3
        System.out.println(stack.pop());        // 2
        System.out.println(stack.pop());        // 1
    }
}