https://www.acmicpc.net/problem/10845
[ 문제 ]
정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.
[ 입력 ]
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다.
둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다.
주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다.
문제에 나와있지 않은 명령이 주어지는 경우는 없다.
[ 출력 ]
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
[ 입출력 예시 ]
예제 입력 | 예제 출력 |
---|---|
15 push 1 push 2 front back size empty pop pop pop size empty pop push 3 empty front | 1 2 2 0 1 2 -1 0 1 -1 0 3 |
문제에 나와있는 push, pop. size, empty, front, back을 메서드를 구현하여 문제를 풀 것이다.
1. static을 이용하여 큐를 나타낼 배열 변수(queue)를 명령의 수가 최대 10000이므로 값도 10000칸 들어올 수 있다. 따라서 크기는 10000으로해서 선언하고,
사이즈를 나타낼 변수(size)와 poll()로 인해서 맨 앞 값이 빠지면 맨 앞을 알려줄 변수(front)와 가장 끝을 알려줄 변수(back)을 0으로 초기화하여 선언한다.
2. push는 값을 큐에 넣어주는 명령이다.
배열 변수 queue에 back에 값을 넣어준다. (size는 poll되는 순간 빠질 수 있으므로)
그럼 size가 늘어났으므로 size를 ++(증가)시켜주고 back값도 ++(증가)시켜준다.
3. pop은 가장 앞에 있는 값을 빼고 출력하는 명령이다.
만약 큐가 비어있다면(size==0)이라면 -1을 반환하고,
그렇지 않다면 제일 앞을 나타내는 queue[front]값을 변수(n)에 담고, 값이 빠졌으므로 size를 --(감소)시키고 front도 한 칸뒤로 이동했으므로 front값을 ++(증가)시켜준다. 그 후 담아뒀던 n을 반환한다.
4. size는 큐의 길이를 출력하는 명령이다.
size값을 반환하여준다.
5. empty는 큐에 값의 유무를 출력하는 명령이다.
배열 queue가 비어있다면(size == 0)이라면 1값을, 그렇지 않다면 0값을 반환한다.
6. front는 가장 맨 앞값을 출력하는 명령이다.
배열이 비어있으면 -1을, 그렇지 않다면 배열 변수 queue의 front인덱스를 반환한다.
7. back은 가장 맨 끝값을 출력하는 명령이다.
배열이 비어있으면 -1을, 그렇지 않다면 배열 변수 queue의 back-1인덱스를 반환한다.
8. 메인 메서드에서 몇 번의 명령을 받을지에 대한 변수(N)을 입력받는다.
9. 명령을 받을 때마다 switch문으로 입력 받은 값에 맞는 코드가 실행되도록 반환 값이 있다면,
StringBuilder(sb)에 담고 개행시켜준다.
10. 최종적으로 StringBuilder(sb)를 출력한다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
static int[] queue = new int[10000];
static int size = 0;
static int front = 0;
static int back = 0;
// push X: 정수 X를 큐에 넣는 연산이다.
public static void push (int val) {
queue[back] = val;
size++;
back++;
}
// pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
public static int pop() {
if(size == 0) {
return -1;
}
else {
int n = queue[front];
size--;
front++;
return n;
}
}
// size: 큐에 들어있는 정수의 개수를 출력한다.
public static int size() {
return size ;
}
// empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
public static int empty() {
if(size == 0) {
return 1;
}
else {
return 0;
}
}
// front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
public static int front() {
if(size == 0) {
return -1;
}
else {
return queue[front];
}
}
// back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
public static int back() {
if(size == 0) {
return -1;
}
else {
return queue[back - 1];
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
StringBuilder sb = new StringBuilder();
int N = Integer.parseInt(st.nextToken());
for(int i=0; i<N; i++) {
st = new StringTokenizer(br.readLine(), " ");
switch(st.nextToken()) {
case "push" :
push(Integer.parseInt(st.nextToken()));
break;
case "pop" :
sb.append(pop()).append("\n");
break;
case "size" :
sb.append(size()).append("\n");
break;
case "empty" :
sb.append(empty()).append("\n");
break;
case "front" :
sb.append(front()).append("\n");
break;
case "back" :
sb.append(back()).append("\n");
break;
}
}
System.out.println(sb);
}
}