[백준] 2617 - 구슬 찾기

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

2617번: 구슬 찾기

 

문제를 처음 봤을 때 구슬... 무게... 순서... 이런 단어가 포함돼 있어서 당연히 위상 정렬일 줄 알았다.

하지만 구하려는 것은 순서가 아닌 정확한 대소 관계를 구해야 했고, 따라서 위상 정렬을 사용할 수 없었다.

 

먼저 절대 무게가 중간이 될 수 없다는 것은 어떻게 구할 수 있을까?

바로 자신보다 큰 무게의 구슬 또는 작은 무게의 구슬의 개수가 (n + 1) / 2 초과일 때 절대로 중간이 될 수 없다.

 

각 구슬의 대소 관계는 입력으로 들어오는 간선과 역 간선을 리스트 형태로 저장하면 구할 수 있다.

즉 모든 정점에 대하여 정간선, 역간선으로 dfs를 수행하고 수행하며 방문한 정점의 개수가 구하려는 값이다.

 

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

int n, m, minV[100], maxV[100], visit[100], rVisit[100];
int v[100][100], rev[100][100];

int getMin(int node) {

	rVisit[node] = 1;
	int ret = 0;

	for (int i = 1; i <= n; i++) {
		if (!rVisit[i] && rev[node][i]) {
			ret += getMin(i);
		}
	}

	return ret + 1;
}

int getMax(int node) {

	visit[node] = 1;
	int ret = 0;

	for (int i = 1; i <= n; i++) {
		if (!visit[i] && v[node][i]) {
			ret += getMax(i);
		}
	}

	return ret + 1;
}

int main() {

	cin >> n >> m;

	for (int i = 0; i < m; i++) {
		int a, b; cin >> a >> b;
		v[a][b] = 1;
		rev[b][a] = 1;
	}

	int ans = 0;
	for (int i = 1; i <= n; i++) {
		int a = getMin(i);
		int b = getMax(i);

		if (a > (n + 1) / 2 || b > (n + 1) / 2)ans++;

		memset(visit, 0, sizeof(visit));
		memset(rVisit, 0, sizeof(rVisit));
	}

	cout << ans;

}