취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 12725번 문제인 Milkshakes (Small)이다.
문제는 아래 링크를 확인하자.

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

12726번 문제에서 입력이 작아진 형태이다. 해당 글을 참고하여 문제를 해결하자.

아래는 제출한 소스코드이다.

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

int total[101]; // [i]: i번 사람이 선호하는 unmalted 가짓수
vector<int> unmalted[11]; // [x]: unmalted x를 좋아하는 사람들
int malted[101]; // [i]: i번 사람이 선호하는 malted의 번호
int ans[11];

void solve() {
	memset(total, 0, sizeof(total));
	for (int i = 1; i < 11; i++) unmalted[i].clear();
	memset(malted, 0, sizeof(malted));
	memset(ans, 0, sizeof(ans));

	queue<int> que;
	int N, M; cin >> N >> M;
	for (int i = 1; i <= M; i++) {
		int T; cin >> T;
		total[i] = T;
		while (T--) {
			int x, y; cin >> x >> y;
			if (y == 0) unmalted[x].emplace_back(i);
			else {
				total[i]--;
				malted[i] = x;
			}
		}
		if (total[i] == 0) que.push(i);
	}

	while (!que.empty()) {
		int cur = que.front(); que.pop();
		if (malted[cur] == 0) {
			cout << "IMPOSSIBLE\n";
			return;
		}
		if (ans[malted[cur]]) continue;
		ans[malted[cur]] = 1;
		for (auto i : unmalted[malted[cur]]) {
			total[i]--;
			if (total[i] == 0) que.push(i);
		}
	}

	for (int i = 1; i <= N; i++) cout << ans[i] << ' ';
	cout << '\n';
}

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

	int T; cin >> T;
	for (int t = 1; t <= T; t++) {
		cout << "Case #" << t << ": ";
		solve();
	}
}