[ Baekjoon ][JAVA][4195] 친구 네트워크

www.acmicpc.net/problem/4195

4195번: 친구 네트워크

문제

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

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

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

입력

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

출력

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

예제입력

예제출력

접근

"유니온파인드"와 "해시맵"을 이용하면 된다.

하지만 특이한점은 노드가 숫자가 아니라 문자열이라는 점이다. 그래서 해시맵을 이용해서 문자열을 키값으로 중복없이 배열을 만들어야 한다.

 

hashmap["이름"] = index 형태로 문자열마다 문자열이 가지는 인덱스를 저장해주었다.

그리고 그 인덱스를 기준으로 parent[] 배열을 만들어 유니온파인드를 사용한다.

 

또한, 여기서 구하고자 하는 값이 관계에 있는 수이기 때문에 유니온하며 크기도 같이 더해준다.

if(u==v) return

에서는 부모가 같으면 이미 유니온 되어 있는 것이기 때문에 이를 하지 않으면 네트워크 크기가 2배가 된다.

 

Union-Find

  -> 유니온 파인드 하면서 개수도 같이 유니온

 

Hash Map

  -> string을 인덱스로 하는 배열  -> 넣으면서 parent[], cnt[] 초기화 해야 시간 효율적!

코드

package baekjoon;

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

public class FriendNetwork_4195 {

	private static int[] parent;
	private static int[] cnt;
	
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuffer sb = new StringBuffer();
		StringTokenizer st;
		HashMap<String, Integer> map;
		int testcase = Integer.parseInt(br.readLine());
		
		while(testcase-- > 0) {
			int F = Integer.parseInt(br.readLine());
			map = new HashMap<>();
			cnt = new int[F*2 + 1];
			parent = new int[F*2 + 1];
			for(int i=1,index=1; i<=F; i++) {
				st = new StringTokenizer(br.readLine());
				String[] strs = {st.nextToken(), st.nextToken()};
				for(int j=0; j<2; j++) {
					if(!map.containsKey(strs[j])) {
						cnt[index] = 1;
						parent[index] = index;
						map.put(strs[j], index++);
					}
				}
				
				int u = map.get(strs[0]);
				int v = map.get(strs[1]);
				union(u, v);
				sb.append(cnt[find(v)]).append("\n");
			}
		}
		System.out.println(sb.toString());
	}
	
	public static void union(int u, int v) {
		u = find(u);
		v = find(v);
		if(u==v) return;
		parent[u] = v;
		cnt[v] += cnt[u];
	}
	
	public static int find(int u) {
		if(parent[u]==u) return u;
		return parent[u] = find(parent[u]);
	}
	

}