하루 하루는 성실하게
인생 전체는 되는대로

[프로그래머스] 파일명 정렬

2 minute read

문제 설명

/* 중략 */

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

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

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

  • 파일명은 우선 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는 함께 입력으로 주어질 수 있다.)

풀이

30분
algorithm헤더의 sort함수를 통해 주어진 조건을 구현하였다. 그런데 기본 예제를 제외한 모든 문제를 틀려서 충격을 받았다. 간단해 보이는 문제에서 이렇게 문제가 틀리는 것은 분명 뭔가 아예 놓치고 있다고 생각이 들어서, 질문게시판을 참고하였고, c++이 제공하는 sort함수가 불안정 정렬인 quicksort를 이용하기 때문이라는 것을 알게 되었다. 아예 내가 정렬하는 코드를 작성해야겠구나 하는 찰나에 c++에서 안정정렬을 함수로 제공해준다는 문구를 보았고, stable_sort를 이용해서 바로 정답을 맞출 수 있었다. 안정/불안정 정렬에 대한 개념도 알고 있었고, c++의 sort는 quicksort 기반이라는 것도 알고 있었지만,

원래 입력에 주어진 순서를 유지한다.

라는 문구를 보고 sort를 그냥 사용하면 안된다는 생각은 미처 하지 못했다.

코드

// 04:26 ~ 04:54
#include <string>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

bool isnum(char c){
    if(c >= '0' && c <= '9') return true;
    return false;
}

bool cmp(string a, string b){
    string head_a = "", head_b = "";
    string num_a = "", num_b = "";
    int idx = 0;
    while(!isnum(a[idx])){
        if(a[idx] >= 'a' && a[idx] <='z'){
            a[idx] = a[idx] - 'a' + 'A';
        }
        head_a += a[idx];
        ++idx;
    }
    while(isnum(a[idx])){
        num_a += a[idx];
        ++idx;
    }
    
    idx = 0;
    while(!isnum(b[idx])){
        if(b[idx] >= 'a' && b[idx] <='z'){
            b[idx] = b[idx] - 'a' + 'A';
        }
        head_b += b[idx];
        ++idx;
    }
    while(isnum(b[idx])){
        num_b += b[idx];
        ++idx;
    }
    
    if(head_a == head_b){
        return stoi(num_a) < stoi(num_b);
    }
    return head_a < head_b;
}

vector<string> solution(vector<string> files) {
    stable_sort(files.begin(), files.end(), cmp);
    return files;
}

You may also enjoy