[백준] 13414 수강신청 - Data Structure / Java

• 문제 링크

13414번: 수강신청

 

• 풀이 과정

순서를 보장하는 Set 자료구조인 LinkedHashSet 을 활용한다.

 

문제에 제시된 2번째 방식에 의해 이미 해당 원소가 존재한다면 삭제 후 삽입하고,

수강 가능 인원 k명 만큼 Set 에서 꺼내 출력한다.

이때 해당 자료구조는 순서를 보장하므로 먼저 삽입된 순서대로 출력되게 된다.

 

• 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Set;
import java.util.StringTokenizer;

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

        int k = Integer.parseInt(st.nextToken());
        int l = Integer.parseInt(st.nextToken());

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

        while (l-- > 0) {
            String input = br.readLine();
            if (set.contains(input))
                set.remove(input);

            set.add(input);
        }

        Iterator<String> it = set.iterator();
        StringBuilder sb = new StringBuilder();

        while (it.hasNext() && k-- > 0)
            sb.append(it.next()).append("\n");

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

}