문제 출처

단계별로 풀어보기 > 심화 2 > 붙임성 좋은 총총이
https://www.acmicpc.net/problem/26069

문제 설명

ChongChong이와 만나면 무지개 댄스에 전염된다.
또한, 무지개 댄스에 전염된 사람과 만나면 무지개 댄스에 전염된다.

사람들의 수 N이 주어지고, 다음 줄부터는 N개의 줄에 걸처 사람들이 만난 기록들이 주어진다.

이 때, 무지개 댄스에 전염 된 사람의 수를 구하라

접근 방법

import java.io.*;
import java.util.HashMap;
import java.util.HashSet;
import java.util.StringTokenizer;

public class 붙임성_좋은_총총이 {

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

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

        HashSet<String> hs = new HashSet<>();
        hs.add("ChongChong");

        for(int i = 0; i<N; i++){
            st = new StringTokenizer(br.readLine());
            String people1 = st.nextToken();
            String people2 = st.nextToken();

            if(hs.contains(people1)){
                hs.add(people2);
            } else if(hs.contains(people2)){
                hs.add(people1);
            }
        }

        bw.write(String.valueOf(hs.size()));
        bw.flush();
        bw.close();
        br.close();

    }

}

Review

import java.io.*;
import java.util.HashSet;
import java.util.StringTokenizer;

public class 붙임성_좋은_총총이_review {

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

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

        HashSet<String> hs = new HashSet<>();

        hs.add("ChongChong");

        for(int i = 0; i<N; i++){
            st = new StringTokenizer(br.readLine());
            String str1 = st.nextToken();
            String str2 = st.nextToken();

            if(hs.contains(str1)){
                hs.add(str2);
            } else if (hs.contains(str2)){
                hs.add(str1);
            }
        }

        bw.write(String.valueOf(hs.size()));
        bw.flush();
        bw.close();
        br.close();
    }
}

알게된 점

문제푼 흔적

Review