[코드그루] 131. 백준 20291번 : 파일 정리

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

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Map;
import java.util.TreeMap;

// 파일 확장자별 정리 몇개씩 있는지 + 사전순 정렬
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());
        Map<String, Integer> list = new TreeMap<>();
        for (int i = 0; i < n; i++) {
            String[] str = br.readLine().split("\\.");
            String file = str[1];
            list.put(file, list.getOrDefault(file, 0) + 1);
        }
        StringBuilder sb = new StringBuilder();
        for (Map.Entry<String, Integer> entry : list.entrySet()) {
            sb.append(entry.getKey()).append(" ").append(entry.getValue()).append("\n");
        }
        System.out.println(sb);
    }
}

쉬운 문제긴 하지만 전보다 시간 줄여서 풀었다 ㅎㅎ

// 이전 풀이
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        Map<String, Integer> map = new HashMap<>();
        for (int i=0; i < N; i++) {
            String[] str = br.readLine().split("\\.");
            String addr = str[1]; // 확장자만 분리
            map.put(addr, map.getOrDefault(addr, 0) + 1); // 확장자 개수 세기
        }
        String[] list = new String[map.size()];
        int num = 0;
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            String ans = entry.getKey() + " " + entry.getValue(); // 확장자 + 개수 문자열 생성
            list[num++] = ans;
        }
        Arrays.sort(list); // 오름차순 정렬
        for(String s : list) { // 배열 for문 돌려서 하나씩 출력
            System.out.println(s);
        }
    }
}