4195번 : 친구 네트워크

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

4195번: 친구 네트워크

 

 

문제

민혁이는 소셜 네트워크 사이트에서 친구를 만드는 것을 좋아하는 친구이다. 우표를 모으는 취미가 있듯이, 민혁이는 소셜 네트워크 사이트에서 친구를 모으는 것이 취미이다.

어떤 사이트의 친구 관계가 생긴 순서대로 주어졌을 때, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.

친구 네트워크란 친구 관계만으로 이동할 수 있는 사이를 말한다.

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스의 첫째 줄에는 친구 관계의 수 F가 주어지며, 이 값은 100,000을 넘지 않는다. 다음 F개의 줄에는 친구 관계가 생긴 순서대로 주어진다. 친구 관계는 두 사용자의 아이디로 이루어져 있으며, 알파벳 대문자 또는 소문자로만 이루어진 길이 20 이하의 문자열이다.

출력

친구 관계가 생길 때마다, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.

 

 

소스코드

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

public class Main {
    static int[] parent, count;

    // 루트 노드를 찾는 find 연산
    static int find(int a) {
        if(a==parent[a])
            return a;
        return parent[a] = find(parent[a]);
    }

    // 두 노드를 같은 집합으로 합치는 Union 연산
    static void union(int a, int b) {
        int rootA = find(a); //a의 부모
        int rootB = find(b); //b의 부모
        //더 작은 값이 부모가 될 수 있도록
        if(rootA>rootB) {
            int temp = rootA;
            rootA = rootB;
            rootB = temp;
        }
        if(rootA == rootB) //부모가 같다면 이미 연결되어 있는 것
            return;
        parent[rootB] = rootA; //b의 부모를 a의 부모로 설정
        count[rootA] += count[rootB];
        return;
    }

    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        int testcase = Integer.parseInt(br.readLine());
        for(int t=1; t<=testcase; t++) {
            HashMap<String, Integer> map = new HashMap<>();
            int N = Integer.parseInt(br.readLine());
            int idx = 0;
            int[][] friend = new int[N][2]; //친구 관계
            for(int i=0; i<N; i++) {
                st = new StringTokenizer(br.readLine());
                String f1 = st.nextToken();
                String f2 = st.nextToken();
                if(!map.containsKey(f1)) {
                    map.put(f1, idx++);
                }
                if(!map.containsKey(f2)) {
                    map.put(f2, idx++);
                }
                friend[i] = new int[] {map.get(f1), map.get(f2)};
            }

            parent = new int[idx];
            //부모 자기 자신으로 설정
            for(int i=0; i<idx; i++) {
                parent[i] = i;
            }

            count = new int[idx];
            Arrays.fill(count, 1);
            for(int[] f: friend) {
                union(f[0], f[1]);
                System.out.println(count[find(f[0])]);
            }
//            for(int i : count) {
//                System.out.println("count = " + i);
//            }
        }
    }
}

 

 

해결방법

혼자서 해결해보려 하였지만, 접근 방식부터 틀려먹었다. "두 사람의 친구 네트워크에 몇 명이 있는지 구하는" 두 사람의 관계에서만 구하는 것인데 친구 전체에서 네트워크에 몇 명이 있는 지 구하는 것으로 문제를 잘못 이해하였다.

문제 해결을 위해 유니온 파인드, 블로그를 참고하였다.

 

 

유니온 파인드라는 그래프 알고리즘을 풀어야 하는 문제였다.

 

풀이방법

1. HashMap을 생성하고 입력으로 새로운 이름이 들어올 때마다 이름을 Key로, Integer값을 Value로 부여한다.
2. Integer값으로 유니온 파인드를 사용하여 친구 관계를 결합한다.
3. 친구 인원 수를 저장하기 위한 별도의 배열을 사용해야 시간초과가 발생하지 않는다. (친구 관계 수 최대 100,000)