[Java] 백준 풀기 20291 - 파일 정리

자바 백준 20291번

실버 3

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

20291번: 파일 정리

 

 

 

 

 

문제 보기

분류: 자료 구조, 문자열, 정렬, 해시를 사용한 집학과 맵, 파싱

 

 

 

 

 

문제 풀기

Array 를 사용할 수 있지만 ArrayList 를 사용해 보고 싶어서 방향을 이쪽으로 잡았다. 해시맵을 사용하지 않아 조건문이 지저분한 경향이 있는데 이번엔 머리에서 떠오르는 대로 답안을 제출하고 해시맵 사용은 다음으로 넘긴다.

 

n 개 만큼 파일 이름을 입력 받으면서 확장자만 분리하여 ArrayList 에 모은다. 이것을 오름차순 정렬을 하여 main 메서드로 넘긴다. main 메서드에서 정렬된 확장자를 두 개씩 비교하면서 같거나 다른 경우 그리고 ArrayList 의 마지막 요소일 경우에 대해 조건문으로 분기하여 출력한다.

 

 

 

 

 

코드 보기

import java.util.Scanner;
import java.util.ArrayList;
import java.util.Collections;

public class Main {
    
    static Scanner sc = new Scanner(System.in);
    
    public static void main(String args[]) {
        
        int n = sc.nextInt();
        
        ArrayList<String> extensionList = sortExtension(n);
        
        int count = 1;
        for (int i = 1; i < extensionList.size(); i++) {
            if ( i == extensionList.size() - 1 && !extensionList.get(i).equals(extensionList.get(i - 1)) ) {
                System.out.println(extensionList.get(i - 1) + " " + count);
                count = 1;
                System.out.println(extensionList.get(i) + " " + count);
            }
            else if ( i == extensionList.size() - 1 && extensionList.get(i).equals(extensionList.get(i - 1)) ) {
                System.out.println(extensionList.get(i) + " " + (count + 1));
            }
            else if ( extensionList.get(i).equals(extensionList.get(i - 1)) ) {
                count += 1;
            }
            else {
                System.out.println(extensionList.get(i - 1) + " " + count);
                count = 1;
            }
        }

    }
    
    public static ArrayList sortExtension (int n) {
        
        ArrayList<String> extensionList = new ArrayList<String>();
        
        for (int i = 0; i < n; i++) {
            String fileExtension = sc.next().split("[.]")[1];
            // System.out.println((fileExtension));  // TEST PRINT
            extensionList.add(fileExtension);
        }
        
        Collections.sort(extensionList);
        // System.out.println(extensionList);  // TEST PRINT
        
        return extensionList;
        
    }
    
}