코딩테스트/TIL

[99클럽] 21일차 문제: 파일 정리

moon101 2025. 2. 17. 20:56

 

 

백준 파일 정리 문제. 확장자 명 순으로 출력하는 문제라 treemap을 사용해서 key로 정렬.

 

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

        int N = Integer.parseInt(st.nextToken()); // 파일 개수

        // 파일 확장자 별로 카운트
        Map<String, Integer> extensionToFreq = new TreeMap<>();

        // 파일 저장
        for (int i = 0; i < N; i++) {
            String s = br.readLine();
            String[] sArr = s.split("\\.");
            
            extensionToFreq.put(sArr[sArr.length -1], extensionToFreq.getOrDefault(sArr[sArr.length -1], 0) + 1);
            
        }

        // 결과 출력
        StringBuilder sb = new StringBuilder();
        for (Map.Entry<String, Integer> e : extensionToFreq.entrySet()) {
            sb.append(e.getKey()).append(" ").append(e.getValue()).append("\n");
        }
        System.out.print(sb);
    }
}