[백준] | JAVA, 자바 | 25192번 - 인사성 밝은 곰곰이

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

 

---

문제 요약 :

ENTER는 새로운 사람이 채팅방에 입장했음을 나타낸다. 그 외는 채팅을 입력한 유저의 닉네임을 나타낸다. 닉네임은 숫자 또는 영문 대소문자로 구성되어 있다.

새로운 사람이 입장한 이후 처음 채팅을 입력하는 사람은 반드시 곰곰티콘으로 인사를 한다. 그 외의 기록은 곰곰티콘을 쓰지 않은 평범한 채팅 기록이다. 

 

( 채팅 기록 중 곰곰티콘이 사용된 횟수를 출력 )

---

문제 풀이 :

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

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());
        HashMap<String, Integer> chat = new HashMap<>();
        int count = 0;
        for (int i = 0; i < n; i++) {
            String s = br.readLine();
            if (s.equals("ENTER")) {
                chat.clear();
            } else {
                if (!chat.containsKey(s)) {
                    chat.put(s, 1);
                    count++;
                }
            }
        }
        System.out.println(count);
    }
}

 

1. 채팅방의 기록 수를 나타내는 정수 n 입력받기

2. HashMap chat 생성, 곰곰티콘이 사용된 횟수를 저장하기 위한 변수 count 생성

3. n 만큼 ENTER 혹은 유저 네임의 문자열 입력받기

3 - 1. 만약 입력받은 문자열이 "ENTER" 라면 chat.clear를 통해 chat를 초기화

3 - 1. 그렇지 않을 경우 chat에 s가 없는지 비교하고 없다면 chat에 s와 1 입력, count 증가

4. count 출력

---

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
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());
        HashSet<String> chat = new HashSet<>();
        int count = 0;
        for (int i = 0; i < n; i++) {
            String s = br.readLine();
            if (s.equals("ENTER")) {
                chat.clear();
            } else {
                if (!chat.contains(s)) {
                    chat.add(s);
                    count++;
                }
            }
        }
        System.out.println(count);
    }
}

 

containsKey를 통해 이미 HashMap에 저장되어 있다면 count나 value값을 증가시키지 않기 때문에 Integer 값 1이 필요가 없다는 생각이 들었다. 그래서 HashMap 대신 ArrayList, HashSet를 사용한 코드로 수정해봤는데 ArrayList는 시간 초과, HashSet은 통과했다. HashMap과의 성능차이는 별로 없는 것 같다.