알고리즘 공부/백준 문제풀이

[백준] 10828 스택 (JAVA)

송테이토 2022. 11. 9. 21:18

백준 10828 자바

 

https://songsunkite.tistory.com/entry/%EC%9E%90%EB%B0%94-%EC%8A%A4%ED%83%9D%EC%9D%B4%EB%9E%80-%EC%9E%90%EB%B0%94-%EC%8A%A4%ED%83%9D-%EA%B5%AC%ED%98%84-STACK 

 

자바 스택이란?- 자바 스택 구현 (STACK)

스택 (STACK)이란? 📌 스택의 개념 스택(stack)이란 쌓아 올린다는 것을 의미한다. 따라서 스택 자료구조라는 것은 책을 쌓는 것처럼 차곡차곡 쌓아 올린 형태의 자료구조를 말한다. 📌 스택의 특

songsunkite.tistory.com

 스택이 어렵다면?

스택

시간 제한 메모리 제한 제출 정답 맞힌 사람 정답 비율

0.5 초 (추가 시간 없음) 256 MB 180414 62864 46043 37.248%

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

  • push X: 정수 X를 스택에 넣는 연산이다.
  • pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
  • size: 스택에 들어있는 정수의 개수를 출력한다.
  • empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
  • top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1

예제 출력 1

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

2
2
0
2
1
-1
0
1
-1
0
3

예제 입력 2

예제 출력 2

7
pop
top
push 123
top
pop
top
pop

-1
-1
123
123
-1
-1

방법1.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

class Stack {

	private int max; // 스택 용량
	private int ptr;// 스택 포인터
	private int[] stk; // 스택본체

	// 실행시 예외 : 스택이 비어있음
	public class EmptyIntStackException extends RuntimeException {
		public EmptyIntStackException() {
		}
	}

	// 실행시 예외 : 스택이 가득 참
	public class OverflowStackException extends RuntimeException {
		public OverflowStackException() {
		}
	}

	// 생성자
	public Stack(int capacity) {
		ptr = 0;// 생성 시 스택은 비어있으므로
		max = capacity;
		try {
			stk = new int[max]; // 스택 본체용 배열 생성
		} catch (OutOfMemoryError e) { // 생성할 수 없음
			max = 0;
		}
	}

	// push X: 정수 X를 스택에 넣는 연산이다.
	public int push(int x) throws OverflowStackException {
		if (ptr >= max)
			throw new OverflowStackException();

		// 전달받은 데이터 x를 배열에 저장하고 스택포인터 증가
		return stk[ptr++] = x;
	}

	// pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다.
	// 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
	public int pop() throws EmptyIntStackException {
		if (ptr <= 0)
			return -1;
		return stk[--ptr];
	}

	// size: 스택에 들어있는 정수의 개수를 출력한다.
	public int size() {
		return ptr;
	}

	// empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
	public int empty() {
		if (ptr <= 0)
			return 1;
		return 0;
	}

	// top: 스택의 가장 위에 있는 정수를 출력한다.
	// 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
	public int top() throws EmptyIntStackException {
		if (ptr <= 0)
			return -1;
		return stk[ptr - 1];
	}
}

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		int N = Integer.parseInt(br.readLine());
		Stack s = new Stack(N);

		StringTokenizer st;

		// 명령어 수만큼 반복.
		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine(), " ");

			// 출력해야하는 명령
			String order = st.nextToken();

			switch (order) {
			case "push":
				s.push(Integer.parseInt(st.nextToken()));
				break;

			case "pop":
				sb.append(s.pop()).append("\\n");
				break;

			case "size":
				sb.append(s.size()).append("\\n");
				break;

			case "empty":
				sb.append(s.empty()).append("\\n");
				break;

			case "top":
				sb.append(s.top()).append("\\n");
				break;

			}
		}
		System.out.println(sb);
	}
}

근데 사실 오류 안던져도 된다.

package Day1109;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

class Stack {

	private int max; // 스택 용량
	private int ptr;// 스택 포인터
	private int[] stk; // 스택본체

	// 생성자
	public Stack(int capacity) {
		ptr = 0;// 생성 시 스택은 비어있으므로
		max = capacity;
		try {
			stk = new int[max]; // 스택 본체용 배열 생성
		} catch (OutOfMemoryError e) { // 생성할 수 없음
			max = 0;
		}
	}

	// push X: 정수 X를 스택에 넣는 연산이다.
	public int push(int x) {
		// 전달받은 데이터 x를 배열에 저장하고 스택포인터 증가
		return stk[ptr++] = x;
	}

	// pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다.
	// 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
	public int pop() {
		if (ptr <= 0)
			return -1;
		return stk[--ptr];
	}

	// size: 스택에 들어있는 정수의 개수를 출력한다.
	public int size() {
		return ptr;
	}

	// empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
	public int empty() {
		if (ptr <= 0)
			return 1;
		return 0;
	}

	// top: 스택의 가장 위에 있는 정수를 출력한다.
	// 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
	public int top() {
		if (ptr <= 0)
			return -1;
		return stk[ptr - 1];
	}
}

public class B10828_1 {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		int N = Integer.parseInt(br.readLine());
		Stack s = new Stack(N);

		StringTokenizer st;

		// 명령어 수만큼 반복.
		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine(), " ");

			// 출력해야하는 명령
			String order = st.nextToken();

			switch (order) {
			case "push":
				s.push(Integer.parseInt(st.nextToken()));
				break;

			case "pop":
				sb.append(s.pop()).append("\\n");
				break;

			case "size":
				sb.append(s.size()).append("\\n");
				break;

			case "empty":
				sb.append(s.empty()).append("\\n");
				break;

			case "top":
				sb.append(s.top()).append("\\n");
				break;

			}
		}
		System.out.println(sb);
	}
}

방법2 .Stack 클래스

package Day1109;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

public class B10828_2 {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		Stack<Integer> stack = new Stack<Integer>();

		int N = Integer.parseInt(br.readLine());

		StringTokenizer st;
		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine(), " ");

			// 출력해야하는 명령
			String order = st.nextToken();

			switch (order) {
			case "push":
				stack.push(Integer.parseInt(st.nextToken()));
				break;

			case "pop":
				sb.append(stack.isEmpty() ? -1 : stack.pop()).append("\\n");
				break;

			case "size":
				sb.append(stack.size()).append("\\n");
				break;

			case "empty":
				sb.append((stack.isEmpty() ? 1 : 0)).append("\\n");
				break;

			case "top":
				sb.append(stack.isEmpty() ? -1 : stack.peek()).append("\\n");
				break;

			}
		}
		System.out.println(sb);
	}
}

시간 차이는 안난다. 개인적으로 Stack클래스가 더 편한듯

코드가 절반으로 줄어드니..