[Java] 백준 20291번 - 파일 정리

문제 링크 : 백준 20291번 - 파일 정리

 

Map을 이용해 확장자명과 개수를 저장해주면 된다.

split을 이용해 .을 기준으로 분할 후 확장자를 key로 지정하고, 이미 key가 있으면 개수를 1개 추가하고 없으면 1개로 넣어주면 된다.

TreeMap이므로 따로 key를 정렬해줄 필요는 없고, 순서대로 iteration으로 key와 value를 함께 출력해주면 된다.

 

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[] input = br.readLine().split("\\.");
    String key = input[1];
    if (list.containsKey(key)) {
        list.put(key, list.get(key) + 1);
    } else {
        list.put(key, 1);
    }
}
for (String s : list.keySet()) {
    System.out.println(s + " " + list.get(s));
}