[Problem Solving - Baekjoon] 4195 친구네트워크

[Baekjoon Online Judge] 4195 친구네트워크

문제

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

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

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

입력

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

출력

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

예제

• input

2
3
Fred Barney
Barney Betty
Betty Wilma
3
Fred Barney
Betty Wilma
Barney Betty

• output

2
3
4
2
2
4

분류

• 자료 구조
• 분리 집합
• 해시를 사용한 집합과 맵

풀이

문제 파악

• 런타임 에러로 실패 -> 성공

• 해시활용
• Union-Find 알고리즘 이용 (합집합 찾기)

2     // testCase
3     // F 친구 관계수
Fred Barney  // 친구관계 연결
Barney Betty
Betty Wilma
3
Fred Barney
Betty Wilma
Barney Betty

• 네트워크 연결되는 순서

구현

전체소스보기

• HashMap 이름을 담을 List를 만들고 이름을 넣음

   static Map<String, Integer> List;
   static int[] Parent, Number;
   static int MAX = 200002;  //사람 수 만 큼 (친구관계수가 100000 을 넘지않고, 한줄에 2명씩들어가서 최대값 잡아줌)

   public static void main(String[] args) throws IOException{
       BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
       BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

       int testCase = Integer.parseInt(br.readLine());

       for (int i = 0; i < testCase; i++) {
           int F = Integer.parseInt(br.readLine());

           List = new HashMap<String, Integer>();
           Parent = new int[MAX];
           Number = new int[MAX];

           int index =0; //입력되는 이름을 index 숫자로 관리

           for (int j = 0; j < F; j++) {
               StringTokenizer st = new StringTokenizer(br.readLine());
               String name1 = st.nextToken();
               String name2 = st.nextToken();

               //입력된 이름 중에 중복여부 체크
               if (!List.containsKey(name1)) {

                   List.put(name1, index);   // 이름을 index로 관리
                   Parent[index] = index;    // 처음에는 부모는 자기자신으로 갖음
                   Number[index] = 1;        // 각 노드가 갖는 연결된 갯수
                   index++;
               }

               if (!List.containsKey(name2)) {
                   List.put(name2, index);
                   Parent[index] = index;
                   Number[index] = 1;
                   index++;
               }

               //각 줄에서 입력 된 이름의 관계성을 찾아 만들어주기 (name1 은 name2의 부모)
               unioun(List.get(name1), List.get(name2));
               bw.write(Number[find(List.get(name1))]+"\n");
           }
       }

       bw.flush();
       bw.close();
       br.close();
   }

• 재귀용법으로 부모를 찾아서 부모 값을 반환 (find : 자기가 속한 집합을 찾음)

public static int find(int x) {
       //부모를 찾아줌
       if ( x == Parent[x]) {
           return x;
       }else {
           int p = find(Parent[x]);
           Parent[x] = p;
           return Parent[x];
       }
   }

• 찾은 부모끼리 비교해서 연산하여 연결 (union : 집합끼리를 합쳐 줌)

public static void unioun(int x, int y) {
    //관계의 갯수를 찾아 줌

    int xR = find(x); //입력된 노드의 부모찾기
    int yR = find(y);

    //부모가 갖지 않으면
    if( xR != yR) {
        //y의 부모를 x라 정해주고
        Parent[yR] = xR;
        //x에 연결된 갯수 = 기존 x에 연결된 갯수 + y에 연결된 갯수
        Number[xR] += Number[yR];
    }
}

---

references

https://www.acmicpc.net/problem/4195 https://soobarkbar.tistory.com/180