백준 | 4195번. 친구 네트워크 - 유니온 & 파인드, HashMap

⬛ 백준 4195번. 친구 네트워크 - 유니온 & 파인드, HashMap

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

4195번: 친구 네트워크

문제

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

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

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

입력

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

출력

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

---

💚나의 풀이

• HashMap<이름, 인덱스> 형태로 선언해서, 각 고유이름별 중복없이 들어오는 순서대로 idx를 부여하도록 했다.
• union, Find로 풀되, 여기서는 각 관계의 정보가 들어온 상태에서 친구관계의 수를 출력해야 하기 때문에 별도의 level[] 배열을 선언했다.
• 기본적으로 최초의 level값은 1로 초기화 (각자 친구는 1명 자기 자신)
• uinon(a, b)는 기본적으로 parent[b]에 a 값을 넣는 구조이기 때문에, level[b]의 친구 관계를 level[a]에 누적하여 합하는 식으로 풀었다.

package to_1011_3;

import java.util.*;
import java.util.Scanner;

/*4195번. 친구 네트워크 - 유니온 파인드 - RE */
public class Main {
	static int TC;
	static int F;
	static int[] parent;
	static int[] level;
	//find
	static int find(int a) {
		if(a == parent[a]) {
			return a;
		}else return parent[a] = find(parent[a]);
	}
	
	//union
	static int union(int a, int b) {
		a = find(a);
		b = find(b);
		if(a!=b) {
			parent[b]= a;
			level[a] += level[b];//b의 친구를 a에 누적시킴
			level[b] = 1;
		}
		return level[a]; //a에 저장된 친구 관계의 수 
	}
	
	
	//실행 메인 
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner kb = new Scanner(System.in);
		
		TC = kb.nextInt();//테케 수
		for(int t=0; t<TC; t++) {
			
			F = kb.nextInt();//친구 관계의 수 
			parent = new int[F*2];
			level = new int[F*2];
			Arrays.fill(level, 1);
			
			for(int i=0; i<F*2; i++) parent[i] = i;
			//이름,, 인덱스 번호 
			HashMap<String, Integer> map = new HashMap<>();
			
			int idx = 0;
			//친구 관계별로 입력받기 
			for(int i=0; i<F; i++) {
				String a = kb.next();
				String b = kb.next();
				if(!map.containsKey(a)) {
					//기존 map에 친구 이름 없는 경우 인덱스를 부여한다.
					map.put(a, idx++);
				}
				if(!map.containsKey(b)) {
					map.put(b, idx++);
				}			
				System.out.println(union(map.get(a), map.get(b)));
			}
		}	
	}
}