[자바]Programmers - 2018 KAKAO BLIND RECRUITMENT [3차] 파일명 정렬

이번에 풀게 된 문제는 "프로그래머스"라는 사이트에서 2018 KAKAO BLIND RECRUITMENT [3차] 파일명 정렬 문제입니다.
Lv. 2에 정답률은 57%인 문제이며 입력 받은 값을 기반으로 정렬하여 알맞은 출력값을 만드는 문제 입니다.

그럼 바로 문제 설명 들어가보도록 하겠습니다.

 

문제 설명

선물을 직접 전하기 힘들 때 카카오톡 선물하기 기능을 이용해 축하 선물을 보낼 수 있습니다. 당신의 친구들이 이번 달까지 선물을 주고받은 기록을 바탕으로 다음 달에 누가 선물을 많이 받을지 예측하려고 합니다.

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 ver-10.zip이 ver-9.zip보다 먼저 표시되기 때문이다.

버전 번호 외에도 숫자가 포함된 파일 목록은 여러 면에서 관리하기 불편했다. 예컨대 파일 목록이 ["img12.png", "img10.png", "img2.png", "img1.png"]일 경우, 일반적인 정렬은 ["img1.png", "img10.png", "img12.png", "img2.png"] 순이 되지만, 숫자 순으로 정렬된 ["img1.png", "img2.png", "img10.png", img12.png"] 순이 훨씬 자연스럽다.

무지는 단순한 문자 코드 순이 아닌, 파일명에 포함된 숫자를 반영한 정렬 기능을 저장소 관리 프로그램에 구현하기로 했다.

소스 파일 저장소에 저장된 파일명은 100 글자 이내로, 영문 대소문자, 숫자, 공백(" "), 마침표("."), 빼기 부호("-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.

파일명은 크게 HEAD, NUMBER, TAIL의 세 부분으로 구성된다.

• HEAD는 숫자가 아닌 문자로 이루어져 있으며, 최소한 한 글자 이상이다.
• NUMBER는 한 글자에서 최대 다섯 글자 사이의 연속된 숫자로 이루어져 있으며, 앞쪽에 0이 올 수 있다. 0부터 99999 사이의 숫자로, 00000이나 0101 등도 가능하다.
• TAIL은 그 나머지 부분으로, 여기에는 숫자가 다시 나타날 수도 있으며, 아무 글자도 없을 수 있다.

파일명	HEAD	NUMBER	TAIL
foo9.txt	foo	9	.txt
foo010bar020.zip	foo	010	bar020.zip
F-15	F-	15	(빈 문자열)

 

파일명을 세 부분으로 나눈 후, 다음 기준에 따라 파일명을 정렬한다.

• 파일명은 우선 HEAD 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. MUZI와 muzi, MuZi는 정렬 시에 같은 순서로 취급된다.
• 파일명의 HEAD 부분이 대소문자 차이 외에는 같을 경우, NUMBER의 숫자 순으로 정렬한다. 9 < 10 < 0011 < 012 < 13 < 014 순으로 정렬된다. 숫자 앞의 0은 무시되며, 012와 12는 정렬 시에 같은 같은 값으로 처리된다.
• 두 파일의 HEAD 부분과, NUMBER의 숫자도 같을 경우, 원래 입력에 주어진 순서를 유지한다. MUZI01.zip과 muzi1.png가 입력으로 들어오면, 정렬 후에도 입력 시 주어진 두 파일의 순서가 바뀌어서는 안 된다.

무지를 도와 파일명 정렬 프로그램을 구현하라.

 

입력 형식

입력으로 배열 files가 주어진다.

• files는 1000 개 이하의 파일명을 포함하는 문자열 배열이다.
• 각 파일명은 100 글자 이하 길이로, 영문 대소문자, 숫자, 공백(" "), 마침표("."), 빼기 부호("-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.
• 중복된 파일명은 없으나, 대소문자나 숫자 앞부분의 0 차이가 있는 경우는 함께 주어질 수 있다. (muzi1.txt, MUZI1.txt, muzi001.txt, muzi1.TXT는 함께 입력으로 주어질 수 있다.)

입출력 예제

입력: ["img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
출력: ["img1.png", "IMG01.GIF", "img02.png", "img2.JPG", "img10.png", "img12.png"]

입력: ["F-5 Freedom Fighter", "B-50 Superfortress", "A-10 Thunderbolt II", "F-14 Tomcat"]
출력: ["A-10 Thunderbolt II", "B-50 Superfortress", "F-5 Freedom Fighter", "F-14 Tomcat"]

 

풀이

우선 정렬이라고 하시면 많은 분들이 Comparable과 Comparator을 구현하여 하실거라고 생각합니다. 저는 아직 많이 써보진 않아서 다소 난감하게 느꼈었습니다. 그리하여 아래에 제가 따로 정리한 글을 첨부해드리겠습니다. 혹시나 저처럼 그렇게 느끼신 분이 계시다면, 아래 글을 먼저 읽어보시고 다시 이 글로 넘어오시는 것을 추천 드립니다.

https://zoohwan99.tistory.com/35

Comparable VS Comparator 의 이해

 

코드

import java.util.Arrays;
import java.util.Comparator;

public class 파일명정렬 {

  public static void main(String[] args) {
    String[] files = {"img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"};
    System.out.println(Arrays.toString(solution(files)));
  }
  public static String[] solution(String[] files) {
    String[] answer = {};
    Arrays.sort(files, new Comparator<String>() {
      @Override
      public int compare(String s1, String s2) {
        String[] file1 = fileInfo(s1);
        String[] file2 = fileInfo(s2);

        int headValue = file1[0].compareTo(file2[0]);//head 부분 비교해주기
        if (headValue == 0) { //head 부분이 같으면
          int numValue1 = Integer.parseInt(file1[1]);
          int numValue2 = Integer.parseInt(file2[1]);
          return Integer.compare(numValue1, numValue2);//number 부분 비교해주기
        }
        return headValue;
      }
      private String[] fileInfo(String s) {
        StringBuilder head = new StringBuilder();
        StringBuilder number = new StringBuilder();
        StringBuilder tail = new StringBuilder();

        int location = 0;

        while (location < s.length() && !Character.isDigit(s.charAt(location))) {//숫자인지 판별하여 숫자가 아닌 문자이면 head에 추가
          head.append(s.charAt(location));
          location++;
        }

        while (location < s.length() && Character.isDigit(s.charAt(location))) {//이 전 while 문에서 문자가 아닌 숫자로 판별되어 while 문 탈출
          number.append(s.charAt(location));                                    //여기서는 숫자인지 판별하여 숫자가 아닌 값이 나올 때까지 number에 추가
          location++;
        }

        if (location < s.length()) {//나머지 부분은 tail에 그냥 다 넣어주기
          tail.append(s.substring(location));
        }

        return new String[]{head.toString().toLowerCase(), number.toString(), tail.toString()};
      }
    });

    answer = files;
    return answer;
  }
}

1. 우선 저는 String[] 배열의 answer을 생성해주었습니다. 그리고 Comparator 인터페이스를 구현한 compare 함수를 객체를 비교해주었습니다.
2. 그 다음 비교해주기에 앞서서 입력 받은 files 를 HEAD, NUMBER, TAIL로 나누어 주기 위해 fileInfo라는 함수를 만들어주었습니다. 이 메서드 안에서는 HEAD, NUMBER, TAIL 별로 StringBuilder를 생성하고 숫자인지 아닌지를 판별해주는 isDigit()을 통해 각각의 형식에 맞게 분류해주었습니다.
3. 그리고 return을 해줄때, 저는 처음에
   return new String[]{head.toString().toLowerCase(), number.toString(), tail.toString()}; 이 아닌
   return new String[]{head.toString().toLowerCase() + number.toString() + tail.toString()};로 해주는 바람에 계속 틀렸다고 나왔습니다.(이 점 유의해주시길..ㅜㅡㅜ)
4. 그렇게 return 받은 갓을 이제 head부분부터 비교해주었고, 만약 head 값이 같다면 number 부분으로 넘어가 또 비교를 해주며 정렬을 진행하였습니다. 그리고 head와 number 부분까지 같다면 원래 입력에 주어진 순서를 유지한다고 하였으므로 tail 부분은 따로 비교하지 않았습니다.

이렇게 정렬 문제를 풀며 아직 Comparable과 Comparator을 많이 안 써본 탓인지 저에겐 아리송한 부분이 존재하였습니다. 그래도 차근차근 문제를 꼼꼼히 읽고 터무니 없는 실수만 하지 않는다면, 다른 분들도 쉽게 풀 수 있으리라 보여집니다.