[Baekjoon] 26043_식당 메뉴

문제(출처: https://www.acmicpc.net/problem/26043)

---

< 식당 메뉴 >

 

문제 풀이 

 

Queue를 활용하여 문제를 해결한다.

유형 1일 때는 Queue에 저장

유형 2일 때는 Queue에서 poll 한 값과 메뉴 번호가 일치하는지 확인 후 일치하면 A에 저장을, 일치하지 않으면 B에 저장한다.

모든 식사가 끝난 뒤 Queue에 값이 남아있다면 C에 저장한다.

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class _26043_ { // 식당 메뉴

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		int n = Integer.parseInt(bf.readLine());

		ArrayList<Integer> a = new ArrayList<>();
		ArrayList<Integer> b = new ArrayList<>();
		ArrayList<Integer> c = new ArrayList<>();

		Queue<int[]> queue = new LinkedList<>();

		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(bf.readLine());

			int cmd = Integer.parseInt(st.nextToken());

			if (cmd == 1) {
				queue.add(new int[] { Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()) });
			} else {
				int num = Integer.parseInt(st.nextToken());

				int info[] = queue.poll();
				if (info[1] == num) {
					a.add(info[0]);
				} else {
					b.add(info[0]);
				}
			}
		}

		while (!queue.isEmpty()) {
			c.add(queue.poll()[0]);
		}

		if (a.size() == 0) {
			bw.write("None\n");
		} else {
			Collections.sort(a);
			for (int i = 0; i < a.size(); i++) {
				bw.write(a.get(i) + " ");
			}
			bw.write("\n");
		}

		if (b.size() == 0) {
			bw.write("None\n");
		} else {
			Collections.sort(b);
			for (int i = 0; i < b.size(); i++) {
				bw.write(b.get(i) + " ");
			}
			bw.write("\n");
		}

		if (c.size() == 0) {
			bw.write("None\n");
		} else {
			Collections.sort(c);
			for (int i = 0; i < c.size(); i++) {
				bw.write(c.get(i) + " ");
			}
			bw.write("\n");
		}
		bw.flush();
	}
}

변수)
n : 정보 개수
a, b, c : 본인이 좋아하는 메뉴를 먹은 학생 목록, 본인이 좋아하지 않는 메뉴를 먹은 학생 목록, 식당에 도착하였으나 식사를 하지 못한 학생 목록
queue : Queue <int []>
cmd : 유형

 

정보 개수를 입력받는다. 정보 개수만큼 정보를 입력받아 다음 과정을 거친다.

 

1) 유형 1이라면 Queue에 저장

2) 유형 2라면 Queue에서 poll 한 후 메뉴 번호를 비교한다. 메뉴 번호가 같다면 a에 저장하고, 같지 않다면 b에 저장한다.

 

모든 식사가 끝난 뒤 Queue에 값이 남아있다면 식사를 하지 못한 학생들이므로 c에 저장한다. 각 ArrayList a, b, c가 비어있다면 None을 출력하고 그렇지 않다면 오름차순으로 정렬한 후 출력한다. 

---

---