백준 10828 자바
스택이 어렵다면?
스택
시간 제한 메모리 제한 제출 정답 맞힌 사람 정답 비율
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클래스가 더 편한듯
코드가 절반으로 줄어드니..
'알고리즘 공부 > 백준 문제풀이' 카테고리의 다른 글
[백준] 1181 : 단어 정렬 (JAVA) (0) | 2022.11.11 |
---|---|
[백준] 10845 : 큐 (JAVA) (0) | 2022.11.10 |
[백준]11650 : 좌표 정렬하기 (JAVA) (0) | 2022.11.09 |
[백준]10867 : 중복 빼고 정렬하기(JAVA) (0) | 2022.11.09 |
[백준] 9093 : 단어 뒤집기 (JAVA) (0) | 2022.11.07 |