취미로 코딩하기

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

이번에 볼 문제는 백준 9466번 문제인 텀 프로젝트이다.
문제는 아래 링크를 확인하자.

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

이 문제에서는 모든 노드의 outdegree가 1인 주어진 그래프에서 사이클에 포함되지 않는 노드의 개수를 세야 한다.

한번의 사이클 검사를 할 때 해당 사이클에 포함된 노드를 전부 처리하는 방식으로 문제를 해결하자.

글쓴이는 visited 배열을 정수로 선언하고, 매번 새로운 사이클 검사를 시작할 때마다 새로운 값으로 방문을 체크하여 방문했던 노드를 만났을 때 이 노드가 이번 탐색에서 새로 체크된(즉 새로이 사이클이 발견되었고 거기에 포함된) 노드인지 이전에 검사된(즉 더 살펴도 처음 노드로 돌아가지 못하는) 노드인지를 판단하였다. 사이클이 발견된 경우, 그 뒤로 visited의 값을 1씩 변화시키면서 사이클을 따라 한 바퀴 더 돌며 사이클에 포함되는 노드의 수를 구할 수 있다.

아래의 구현에서는 미리 전체 노드의 개수에서 빼 결과적으로 사이클에 포함되니 않는 노드의 수가 남게끔 하였다.

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

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

int G[100001];
int visited[100001];

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

	int T; cin >> T;
	while (T--) {
		memset(G, 0, sizeof(G));
		memset(visited, 0, sizeof(visited));
		int N; cin >> N;
		int ans = N;
		int mark = 1;
		for (int i = 1; i <= N; i++) cin >> G[i];
		for (int i = 1; i <= N; i++) {
			if (visited[i]) continue;
			int current = i;
			while (!visited[current]) {
				visited[current] = mark;
				current = G[current];
			}
			while (visited[current] == mark) {
				ans--;
				visited[current]++;
				current = G[current];
			}

			mark += 2;
		}

		cout << ans << '\n';
	}
}