반응형
문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
- push X: 정수 X를 스택에 넣는 연산이다.
- pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 스택에 들어있는 정수의 개수를 출력한다.
- empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
- top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
예제 입력 1
14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top
예제 출력 1
2
2
0
2
1
-1
0
1
-1
0
3
나의 풀이
이 문제는 그냥 스택을 만들면되는 문제였다. 스택을 처음 접해보는 사람은 어려울수도 있겠지만 어느정도 알고리즘 문제를 접해봤다면 아주 쉽게 풀 수 있는 문제 중 하나이다.
우선
List<Integer> stack = new Arraylist()
를 사용하여 스택을 만들어준다. ( 물론 이 문제에서는 N을 알려주기 때문에 최대 크기인 14만큼의 int형 배열을 사용해도 되지만 필자는 크기가 정적인 배열보다는 List가 동적으로 사이즈가 조절되고 size를 구해주는 메서드도 있기때문에 List를 사용하였다. )그리고나서는 각 명령마다 메서드를 만들어주었다.
push(int num)
: 아규먼트로 스택에 넣을 숫자를 입력받고 List에add()
를 사용하여 맨 뒤에 넣어준다.pop()
: 스택은후입선출
이므로 최근에 들어온게 pop되어야한다. push에서 맨 뒤에 넣어주었으니 맨 뒤의 인자를 반환함과 동시에 스택에서remove()
를 사용하여 삭제해준다.sizeOf()
: int형 배열을 사용한다면 for문을 돌려 size를 구할 수 있고, List를 사용할 경우size()
를 사용하면 쉽게 구할 수 있다.isEmpty()
:size()
를 구해서0이면 true
,0이 아니면 false
를 반환해주어도 되지만 간단하게 List의isEmpty()
를 사용했다.top()
: top는 pop과 비슷하지만 삭제는 되지않는 명령이니까remove
를 하지않고get()
으로 값만 반환해준다.
코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;
/**
* 10828번 스택
*/
public class Baekjoon10828 {
static List<Integer> stack = new ArrayList<>();
static StringTokenizer st = null;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
for (int i=0; i<n; i++) {
st = new StringTokenizer(br.readLine());
String op = st.nextToken();
if (op.equals("push")) push(Integer.parseInt(st.nextToken()));
else if (op.equals("pop")) System.out.println(pop());
else if (op.equals("size")) System.out.println(sizeOf());
else if (op.equals("empty")) {
if (isEmpty()) System.out.println(1);
else System.out.println(0);
}
else if (op.equals("top")) System.out.println(top());
else System.out.println("Error");
}
}
private static void push(int num) {
stack.add(num);
}
private static int pop() {
if (isEmpty()) return -1;
else return stack.remove(stack.size() - 1); // remove()는 삭제가 되는 인덱스의 값을 반환과 동시에 삭제를 한다.
}
private static int sizeOf() {
return stack.size();
}
private static Boolean isEmpty() {
return stack.isEmpty();
}
private static int top() {
if (isEmpty()) return -1;
else return (stack.get(stack.size()-1));
}
}
반응형
'Programming > Algorithm' 카테고리의 다른 글
[백준9012번] 괄호 (0) | 2020.01.24 |
---|---|
[백준10773번] 제로 (0) | 2020.01.24 |
[백준1676번] 팩토리얼 0의 개수 (0) | 2020.01.19 |
[백준9375번] 패션왕 신해빈 (0) | 2020.01.19 |
[백준11051번] 이항 계수 2 (0) | 2020.01.19 |