백준 20291 - 파일 정리

문제

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

20291번: 파일 정리

 

 

풀이

확장자만 가져오기 위해 substring 메소드를 사용한다. 

확장자를 오름차순으로 정렬 후 앞에서 부터 개수를 세면 된다.

 

 

코드

 

java

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    static int N;
    static String[] arr;
    static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) {
        input();
        solve();
    }

    private static void solve() {
        Arrays.sort(arr);
        int cnt = 1;
        for (int i = 1; i < N; i++) {
            if (arr[i].equals(arr[i - 1])) {
                cnt++;
            } else {
                sb.append(arr[i - 1] + " " + cnt + "\n");
                cnt = 1;
            }

            if (i == N - 1)
                sb.append(arr[i] + " " + cnt + "\n");
        }
        System.out.println(sb.toString());
    }

    private static void input() {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        arr = new String[N];
        for (int i = 0; i < N; i++) {
            String str = sc.next();
            arr[i] = str.substring(str.indexOf('.') + 1);
        }

        sc.close();
    }
}