백준 7785 회사에 있는 사람 [JAVA]

Ga0·2023년 5월 10일

baekjoon

목록 보기

45/125

문제 해석

첫번째 줄에 출입 기록 수(N)을 입력받고, 출입 기록 수(N)만큼 출입 기록이 순서대로 주어진다.
그 출입 기록은 출입한 사람이름과 enter(출근) or leave(퇴근)이 주어지는데, 현재 회사에 남아있는 사람을 출력하면 되는 문제이다.
다시 말해 퇴근하지 않은 모든 사람의 이름을 출력하면된다. 단, 사람의 이름을 사전 순의 역순으로 한 줄에 한명식 출력해야한다.

틀린 코드

import java.io.*;
import java.util.*;

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

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

        //해시 맵으로 정렬하면 키 값이 중복으로 들어갈 수 없기 때문에 문제 풀기 편하다.
        HashMap<String, String> workArray = new HashMap<String, String>();

        for (int i = 0; i < N; i++){
            StringTokenizer st = new StringTokenizer(br.readLine());
            //만약 baha가 enter로 출근 하고 leave로 퇴근했다면 baha가 두번 생기는 것이 아니라 baha=leave로 갱신된다.
            workArray.put(st.nextToken(), st.nextToken()); //키 값이 중복되면 새로운 값으로 갱신될 것이다.
            //즉, 이름은 key값으로 출퇴근 여부를 value값으로 넣는다.
        }

        br.close();

        List<String> keyset = new ArrayList<String>(workArray.keySet()); //역순 정렬을 위해
        Collections.reverse(keyset); //역순 정렬

        for(String key : keyset){
            if(workArray.get(key).equals("enter")){ //갱신되기 때문에 enter일 경우만 남아있는 경우
                bw.write(key + "\n"); //사람의 이름을 출력해야하므로 key값을 넣었다.
            }
        }

        bw.flush();
        bw.close();
    }

}

해시맵을 사용하면 간단하게 풀릴 것 같아서 코드를 이렇게 작성했지만...

틀린 결과

틀렸다...😢

맞은 코드

import java.io.*;
import java.util.*;

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

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

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

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

            String key = st.nextToken(); //키 (사람 이름)
            String value = st.nextToken(); // 값(출퇴근 여부)

            if(value.equals("enter")){ //출근일 경우 HashSet에 저장하고(WorkArray)
                workArray.add(key);
            }else if(value.equals("leave")){ //퇴근일 경우 HashSet에서 지운다.
                workArray.remove(key);
            }
        }

        br.close();

        //정렬을 위해 list를 따로 만들어줌(Collections.sort()위해)
        ArrayList<String> list = new ArrayList<String>(workArray);
        Collections.sort(list);

        for(int i = list.size()-1;  i >= 0; i--){ //역순 정렬을 위해 뒤부터 출력
            bw.write(list.get(i) + "\n");
        }

        bw.flush();
        bw.close();
    }

}

hashmap을 사용했을 경우 왜 틀렸는지는 모르겠지만, 추측으로는 시간초과인 것 같다.(map을 사용할 경우 key의 검색 속도가 속도를 좌우한다. 하지만 hashset은 key값이 따로 존재하지 않으므로 시간이 덜 들지 않을까라는 생각...?) 잘 모르겠지만...
그래서 hashset을 사용하였고, 퇴근의 경우를 값을 갱신하는 것이 아닌 해당 값이 퇴근(leave)이면 hashset에서 삭제해 주었다. (반복문이 덜 돌 것이다. 풀로 저장하는 것 보다)