[Java] 백준알고리즘 #26069 붙임성 좋은 총총이

백준알고리즘 #26069 붙임성 좋은 총총이

백준알고리즘 #26069 붙임성 좋은 총총이

 

사람들은 두명씩 만나는데, 무지개 춤을 추는 사람을 만나면 이 사람은 이후 무지개댄스를 춘다.

이 때, 무지개 춤을 추는 사람의 수를 구하는 문제.

 

앞선 인사성 밝은 곰곰이와 비슷한 문제이다.

2023.10.17 - [Java/백준알고리즘] - [Java] 백준알고리즘 #25192 인사성 밝은 곰곰이

 

이 역시 HashSet을 이용하면 간단하게 문제를 해결할 수 있다.

2023.10.17 - [Java/개념정리] - [Java] HashSet 에 대해 알아보자

 

 

먼저 초기값을 지정하여 (무지개 춤을 추는 ChongChong) Hash를 생성한다.

HashSet<String> user = new HashSet<String>(Arrays.asList("ChongChong"));

 

이후 반복문을 통해 입력을 받고, 입력받은 문자열(사람)을 공백을 기준으로 잘라낸다.

잘라낸 문자열(사람1, 사람2)이 각각 위 Hash에 포함되어 있는지 확인하고, 한 명이라도 포함되어 있다면 둘다 Hash에 추가한다.

for (int i = 0; i < n; i++) {
    String input = br.readLine();
    String userA = input.split(" ")[0];
    String userB = input.split(" ")[1];
    if (user.contains(userA) || user.contains(userB)) {
        user.add(userA);
        user.add(userB);
    }

}

 

마지막까지 반복한 후 이 Hash의 크기를 반환해주면 완성.

 

전체 코드는 다음과 같다.

 

// 해설참조 : sehyeok.tistory.com 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.HashSet;

public class Main {
    public static void main(String args[]) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        
        // Hash생성 (초기값 : ChongChong)
        HashSet<String> user = new HashSet<String>(Arrays.asList("ChongChong")); 

        for (int i = 0; i < n; i++) {
            String input = br.readLine();
            String userA = input.split(" ")[0];		// 사람1
            String userB = input.split(" ")[1];		// 사람2
            if (user.contains(userA) || user.contains(userB)) {
                user.add(userA);
                user.add(userB);
            }

        }
        System.out.println(user.size());		// Hash의 크기 출력

        br.close();
    }

}