[BOJ][9466] 텀 프로젝트

1. 문제 링크

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

2. 사용 알고리즘

DFS

3. 풀이

1) 아래 세 개의 배열 선언

1. 어느 정점과 연결되는지 저장하는 배열 arr

2. DFS 탐색의 순서를 저장하기 위한 배열 check

3. DFS 탐색의 시작 정점을 저장하기 위한 배열 start

2) dfs 정의 - int dfs(int num, int s, int cnt)

• num : 현재 정점

• s : 현재 진행중인 DFS 탐색의 시작 정점

• cnt : 현재 정점까지 이동한 횟수

3) 1 ~ N번까지 아직 방문하지 않은 정점(check[i] == 0)에 대하여 DFS 탐색 시작

• 아직 방문하지 않은 정점인 경우(check[i] == 0)

  • DFS 탐색 계속 진행

• 이미 방문한 정점인 경우(check[i] != 0)

  • start[i]와 현재 DFS 탐색의 시작 정점이 같은 경우

    • 현재까지 탐색 횟수 - check[i] 리턴

  • start와 현재 DFS 탐색의 시작 정점이 다른 경우

    • 0 리턴

4) DFS 탐색을 통해 팀에 속한 학생의 수 확인 완료

∴ 총 학생수 - DFS 탐색을 통해 구한 학생 수

4. 소스 코드

4-1. C++

https://github.com/dev-aiden/problem-solving/blob/main/boj/9466.cpp

#include <iostream>

using namespace std;

int arr[100003];
int check[100003];
int start[100003];

int dfs(int num, int s, int cnt) {
    if(check[num]) {
        if(start[num] == s) return cnt - check[num];
        return 0;
    }
    check[num] = cnt;
    start[num] = s;
    return dfs(arr[num], s, cnt + 1);
}

int main(void) {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    int t; for(cin >> t; t--;) {
        int n; cin >> n;
        for(int i = 1; i <= n; ++i) {
            cin >> arr[i];
            start[i] = check[i] = 0;
        }
        int ans = 0;
        for(int i = 1; i <= n; ++i) {
            if(!check[i]) ans += dfs(i, i, 1);
        }
        cout << n - ans << "\n";
    }
    return 0;
}

4-2. JAVA

https://github.com/dev-aiden/problem-solving/blob/main/boj/9466.java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    
    static int arr[] = new int[100003];
    static int check[] = new int[100003];
    static int start[] = new int[100003];

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();
        int t; for(t = Integer.parseInt(br.readLine()); t-- > 0;) {
            int n = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            for(int i = 1; i <= n; ++i) {
                arr[i] = Integer.parseInt(st.nextToken());
                start[i] = check[i] = 0;
            }
            int ans = 0;
            for(int i = 1; i <= n; ++i) {
                if(check[i] == 0) ans += dfs(i, i, 1);
            }
            sb.append(n - ans).append("\n");
        }
        System.out.println(sb);
    }

    public static int dfs(int num, int s, int cnt) {
        if(check[num] != 0) {
            if(start[num] == s) return cnt - check[num];
            return 0;
        }
        check[num] = cnt;
        start[num] = s;
        return dfs(arr[num], s, cnt + 1);
    }
}