[백준 28107] 회전초밥 (C++)

https://www.acmicpc.net/problem/28107

문제

---

회전 초밥 가게에 N$N$명의 손님이 있고, 요리사는 M개의 초밥을 순서대로 만든다. 요리사가 초밥을 만들 경우, 1번 손님부터 N번 손님의 순서대로 그 초밥을 받게 된다. 만약 먼저 초밥을 받는 손님이 초밥을 먹을 경우, 뒤의 손님들은 해당 초밥을 먹을 수 없다. 만약 아무도 해당 초밥을 먹지 않는다면, 초밥은 버려진다.

 N명의 손님은 각자 먹고 싶은 초밥이 적힌 주문 목록을 가지고 있다. 목록에 적힌 초밥의 순서에 상관 없이 만약 목록에 적혀있는 초밥이 앞에 오면 반드시 먹는다. 만약, 목록에 적히지 않은 초밥을 받는다면 그 초밥은 반드시 먹지 않는다. 단, 손님들은 다양한 초밥을 먹고 싶어하기 때문에 각 종류의 초밥은 최대 한 번만 먹는다.

각 손님의 주문 목록과 순서대로 만들어지는 M개의 초밥이 주어질 때, 각 손님이 먹게 되는 초밥의 개수를 구해 보자.

 

입력

---

첫 번째 줄에 손님의 수 N과 초밥의 수 M이 공백으로 구분되어 주어진다. (1 ≤ N ≤ 100 000;1 ≤ M ≤ 200000)

두 번째 줄부터 N개의 줄에 걸쳐 각 손님에 대한 주문 목록을 나타내는 정수 k와 A1,A2,…,Ak가 공백으로 구분되어 순서대로 주어진다. k는 주문 목록에 적힌 초밥 종류의 개수이며, Ai는 주문 목록에 적힌 초밥 종류이다. (1 ≤ Ai ≤ 200000)

각 손님의 주문 목록에 적힌 초밥 종류는 모두 다르며, 모든 손님에 대해 k의 합이 200000 이하임이 보장된다.

 N+2번째 줄에 요리되는 초밥의 종류를 나타내는 M개의 정수 B1,B2,…,B이 공백으로 구분되어 순서대로 주어진다. (1 ≤ Bi ≤ 200000)

 

출력

---

한 줄에 N개의 정수 C1,C2,…CN을 공백으로 구분하여 출력한다. Ci는 i번 손님이 먹은 초밥의 개수를 의미한다.

 

문제 풀이

---

큐를 사용하는 문제.

 

각 초밥 종류에 대해 큐를 만들고, 손님의 주문 목록을 입력받을 때 해당 초밥 번호에 대한 큐에 손님의 번호를 push한다. 이후 만든 초밥의 종류를 입력받을 때 번호에 해당하는 큐에서 pop해준다. 큐에 들어 있는 번호는 손님의 번호이므로 이를 인덱스 삼아 해당 손님이 먹은 초밥의 개수를 기록하는 배열의 값을 1 증가시켜주면 된다.

 

아래는 코드.

#include <iostream>
#include <queue>
using namespace std;

int main()
{
	cin.tie(NULL);
	ios::sync_with_stdio(false);

	int N, M, T, num;
	cin >> N >> M;
	queue<int>* q = new queue<int>[200001];
	int* sushiCounts = new int[N];
	for (int i = 0; i < N; i++)
	{
		cin >> T;
		sushiCounts[i] = 0;
		for (int j = 0; j < T; j++)
		{
			cin >> num;
			q[num].push(i);
		}
	}
	for (int i = 0; i < M; i++)
	{
		cin >> num;
		if (!q[num].empty())
		{
			int current = q[num].front();
			sushiCounts[current]++;
			q[num].pop();
		}
	}
	for (int i = 0; i < N; i++)
	{
		cout << sushiCounts[i]<<" ";
	}
	return 0;
}