본문 바로가기
코딩테스트 준비/백준

[백준] 13414번 - 수강신청 [Java]

by mwzz6 2024. 12. 23.

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

 

[백준] 13414번 - 수강신청 [Java]
[백준] 13414번 - 수강신청 [Java]


1.  아이디어

 

LinkedHashSet 자료구조를 활용하면 간단하게 해결할 수 있다.


2. 문제풀이

 

수강신청 과정은 순서가 있고 성공, 실패 두 가지로 나뉜다는 점에서 이 두 가지를 한번에 다룰 수 있는 LinkedHashSet 자료구조를 이용했다. 버튼을 누른 순서로 대기목록에 들어가고 버튼을 한번 더 누르면 대기목록의 맨 뒤로 밀려나는건 LinkedHashSet에 존재하지 않는 학번이면 넣고 존재하는 학번이면 LinkedHashSet의 학번을 제거하고 새로 넣는 방식으로 구현할 수 있다. 일반 HashSet은 순회 과정에서 넣은 순서로 나오지 않아서 사용하지 않았고, 학번은 앞자리가 0인 경우가 있어서 String 타입으로 받아서 처리했다.


3. 코드

 

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));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        int K = Integer.parseInt(st.nextToken());
        int L = Integer.parseInt(st.nextToken());

        Set<String> set = new LinkedHashSet<>();

        for (int i = 0; i < L; i++) {
            String id = br.readLine();

            set.remove(id);
            set.add(id);
        }

        int cnt = 0;
        for (String id : set) {
            sb.append(id).append("\n");
            cnt++;

            if (cnt == K) break;
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

4. 후기