백준 4195.

문제이름 친구 네트워크

Featured image

백준 문제 4195 친구 네트워크

친구 네트워크 백준 4195

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

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

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

입력

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

출력

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

// Example can be run directly in your JavaScript console

// Create a function that takes two arguments and returns the sum of those arguments
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.StringTokenizer;

public class Main {

	static int F,parent[],cnt[];
	static StringTokenizer st;
	public static void main(String[] args) throws IOException{
		BufferedReader br= new BufferedReader(new InputStreamReader(System.in));
		int testcase = Integer.parseInt(br.readLine());
		for(int t= 0 ; t < testcase; t++) {
			int num = 0 ,idx1=0,idx2=0;
			F = Integer.parseInt(br.readLine());
			ArrayList<Node> arr = new ArrayList<>();
			 parent = new int[200000];
			 cnt = new int[200000];
			HashMap<String,Integer> index =  new HashMap<String,Integer>();

			for(int i=0;i<parent.length;i++) {
				parent[i]=i;
				cnt[i]=1;
			}								//makeSet;
			int count = 0;
			for(int f = 0 ; f< F; f++) {	//친구관계 수만큼 돌리기
				st = new StringTokenizer(br.readLine());
				String str1 = st.nextToken();
				String str2 = st.nextToken();
				if(!index.containsKey(str1)) {
					index.put(str1, num++);
				}
				idx1 = index.get(str1);
				if(!index.containsKey(str2)) {
					index.put(str2,num++);
				}
				idx2 = index.get(str2);


//				boolean flag1 = false,flag2 = false;
//				st = new StringTokenizer(br.readLine());
//				Node node1 = new Node(st.nextToken(),count++);		//친구 1번
//				Node node2  = new Node(st.nextToken(),count++);		//친구 2번
//				for(int i = 0 ; i< arr.size(); i++) {				//어레이에 이미 등록되어있는 사람인지 확인
//					if(arr.get(i).name.equals(node1.name)) {
//						node1 = new Node(arr.get(i).name,arr.get(i).index);
//						flag1 = true;
//					}
//					else if(arr.get(i).name.equals(node2.name)) {
//						node2 = new Node(arr.get(i).name,arr.get(i).index);
//						flag2 = true;
//					}
//				}
//				if(!flag1)arr.add(node1);	//등록안되어있으면 등록
//				if(!flag2)arr.add(node2);


				unionSet(idx1,idx2);		//합집합
				//System.out.println(cnt[findSet(node1.index)]);
			}
		}

	}
	private static void unionSet(int idx1, int idx2) {
		int x = findSet(parent[idx1]);
		int y = findSet(parent[idx2]);
		if(x!=y) {
			parent[y]=x;
			cnt[x] = cnt[x]+cnt[y];
			cnt[y] = cnt[y]-cnt[y];
		}
		System.out.println(cnt[x]);
	}
	private static int findSet(int x) {
		if(parent[x]==x) {
			return x;
		}else {
			return parent[x] = findSet(parent[x]);
		}
	}
	static class Node{
		String name;
		int index;

		Node(String name,int index){
			this.name = name;
			this.index = index;
		}
	}

}


// Call the function

// > 8