알고리즘 문제풀이백준-1043 C++

이동복·2022년 1월 14일

🎲문제

지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.

사람의 수 N이 주어진다. 그리고 그 이야기의 진실을 아는 사람이 주어진다. 그리고 각 파티에 오는 사람들의 번호가 주어진다. 지민이는 모든 파티에 참가해야 한다. 이때, 지민이가 거짓말쟁이로 알려지지 않으면서, 과장된 이야기를 할 수 있는 파티 개수의 최댓값을 구하는 프로그램을 작성하시오.

⌨ 입력

첫째 줄에 사람의 수 N과 파티의 수 M이 주어진다.

둘째 줄에는 이야기의 진실을 아는 사람의 수와 번호가 주어진다. 진실을 아는 사람의 수가 먼저 주어지고 그 개수만큼 사람들의 번호가 주어진다. 사람들의 번호는 1부터 N까지의 수로 주어진다.

셋째 줄부터 M개의 줄에는 각 파티마다 오는 사람의 수와 번호가 같은 방식으로 주어진다.

N, M은 50 이하의 자연수이고, 진실을 아는 사람의 수는 0 이상 50 이하의 정수, 각 파티마다 오는 사람의 수는 1 이상 50 이하의 정수이다.

✅ 풀이

풀이방법

입력값이 매우 적기 때문에 시간에 제약이 거의 없이 풀 수 있다. bfs와 비슷한 방식으로 해결했다. 큐에 초기 진실을 아는 사람의 번호들을 집어넣고, 반복문을 돌려가면서, 파티 내에 진실을 아는 사람이 있다면 이미 검사한 번호 인지 체크하여(visited) 큐에 집어 넣는다. 큐가 빌 때 까지 반복하며, 마지막에 진실을 아는 사람이 없는 파티들을 출력 한다.

N과 M, 진실을 아는 사람의 수, 정보를 입력받는다 이때 진실을 아는사람의 수를 큐에 push한다.

전체 파티의 정보를 배열에 입력받는다.

void init() {
	cin >> N >> M >> tmp;
	
	for (int i = 0; i < tmp; i++) {
		int num;
		cin >> num;
		q.push(num);
	}
	for (int i = 1; i <= M; i++) {
		int num;
		cin >> num;
		for (int j = 0; j < num; j++) {
			cin >> tmp;
			party[i][tmp] = true;
		}
	}
}

큐가 빌때 까지 반복하는 while문을 생성하고 파티 내에 큐의 가장 앞 부분 숫자(진실을 아는 사람)이 존재할 경우, 해당 파티의 모든 사람중 진실을 아는 사람이 아니였던 사람들에 한하여 큐에 추가한다.
사용되지 않는 해당파티의 0번 인덱스를 true 처리하여, 이미 순회한 파티임을 표시한다.
마지막으로 큐의 가장 앞부분을 pop 한 후, 큐가 빌 때 까지 반복한다.

전체 파티의 0번째 인덱스가 false인 경우를 세어 출력한다.

void solve() {
	while (!q.empty()) {
		for (int i = 1; i <= M; i++) {
			if (party[i][0] == false && party[i][q.front()] == true) {
				party[i][0] = true;
				visited[q.front()] = true;

				for (int k = 1; k <= N; k++) {
					if (party[i][k] == true && visited[k] == false) {
						q.push(k);
					}
				}
			}
		}
		q.pop();
	}
	
	for (int i = 1; i <= M; i++) {
		if (party[i][0] == false) {
			result++;
		}
	}
	cout << result;
}

전체 코드

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <functional>

using namespace std;

int M, N;
int tmp, result = 0;
bool party[51][51] = { 0, };
bool visited[51] = { 0, };
queue<int> q;

void init();
void solve();

void init() {
	cin >> N >> M >> tmp;
	
	for (int i = 0; i < tmp; i++) {
		int num;
		cin >> num;
		q.push(num);
	}
	for (int i = 1; i <= M; i++) {
		int num;
		cin >> num;
		for (int j = 0; j < num; j++) {
			cin >> tmp;
			party[i][tmp] = true;
		}
	}
}

void solve() {
	while (!q.empty()) {
		for (int i = 1; i <= M; i++) {
			if (party[i][0] == false && party[i][q.front()] == true) {
				party[i][0] = true;
				visited[q.front()] = true;

				for (int k = 1; k <= N; k++) {
					if (party[i][k] == true && visited[k] == false) {
						q.push(k);
					}
				}
			}
		}
		q.pop();
	}
	
	for (int i = 1; i <= M; i++) {
		if (party[i][0] == false) {
			result++;
		}
	}
	cout << result;
}

int main() {
	init();
	solve();
	return 0;
}

이동복

아는것 하나 없는 유기물 덩어리

이전 포스트

알고리즘 문제풀이백준-18870 C++

다음 포스트

알고리즘 문제풀이백준-1043 C++