본문 바로가기
파이썬/코딩테스트

[파이썬]코딩테스트 06 정렬_파일명

by nemonemonemo 2025. 8. 5.

문제

 

프로그래머스

SW개발자를 위한 평가, 교육의 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr


문제 설명

파일명 정렬

세 차례의 코딩 테스트와 두 차례의 면접이라는 기나긴 블라인드 공채를 무사히 통과해 카카오에 입사한 무지는 파일 저장소 서버 관리를 맡게 되었다.

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 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"]

</aside>


풀이

  • 문제 상황을 체크
  • 기존 정렬 방식 : 글자별로 코드값을 기준으로 한 자리씩 정렬 ⇒ 10, 2 꼬인다
  • 파일명
    • 영문 대소, 숫자, 공백, 마침표, -
    • 문자열 길이 최대 100글자 → 최소) 영문 1개+ 숫자1개 ⇒ 2글자(영+숫) ~ 100글자 → [연속은 5자리가 max + 0] ….[연속숫자 5자리 max + 0]
  • 파일명 구성 : 3파트
    • head
    • number
    • tail
  • 위의 3가지 파트 정보 바탕으로 정렬 기준
    • 정렬1기준
      • head파트 & 사전순(오름차순)
      • 단, 대소구별 x → 통일하면 편하지 않을까
    • 정렬2기준
      • number파트 &숫자크기(오름차순)
      • 단, 앞에 zero padding에 대한 처리 —> 0010⇒ 10
    • 정렬3기준
      • 원본 순서 유지
      • tail은 필요 없네
      • 대신 필요 정보에 원본 순서가 있으면 좋을 듯
  • 주관적인 결론
    • 정렬에 대한 필요한 정보 (head, number, 원본순서)
    • +최종출력에는 → 원본파일명 필요
    • (원본파일명, head, number, 원본순서) ← 이렇게 구성하는 것은 개취

# 샘플 케이스 분석,,
#입력: ["img12.png", "img10.png", "img02.png", "img1.png",
#  "IMG01.GIF", "img2.JPG"]
# --> 출력: ["img1.png", "IMG01.GIF", "img02.png", "img2.JPG", "img10.png", "img12.png"]

#1) 필요한 정보 추출 : head, number, 원본순서
"img12.png"         :   img    12     0
"img10.png"         :   img    10     1
"img02.png"         :   img    02     2
"img1.png"          :   img     1     3
"IMG01.GIF"         :   IMG    01     4
"img2.JPG"          :   img     2     5
# --> number 기준으로 앞 headm, 뒤에 tail--> 안씀!!!
#     number를 찾는 규칙 => 문자열에 대한 규칙[정규식] 
#                        -> 정규식은 데이터수집/전처리 할 때 씀
# ex2)
#입력: ["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"]
#
# "F-5 Freedom Fighter" --> F-, 5,  0    (3)
# "B-50 Superfortress"  --> B-, 50, 1    (2)
# "A-10 Thunderbolt II" --> A-, 10, 2    (1)
# "F-14 Tomcat"         --> F-, 14, 3    (4)

 

정규식

  • 정규식에 대한 내용은 좀 알아야한다
 

08-2 정규 표현식 시작하기

정규 표현식에서는 메타 문자(meta characters)를 사용한다. 먼저 메타 문자가 무엇인지 알아보자. [TOC] ## 정규 표현식의 기초, 메타 문자 메타 문자란 원…

wikidocs.net

  • 예전: 직접 룰을 보며, 검색, 직접 규칙 만들어서 ⇒ 룰 베이스 + 예전부터 많이 사용 : gpt 잘 함
  • 코테 준비하는 입장: 기본적인 내용을 파악하고 쓸 정도

문제를 풀기위한 사전지식

  • 사전지식1) 정규식
    • 목적
      • ’문자열’ 내에서 내가 원하는 패턴을 찾는 룰 (규칙)
      • → 일반적인 모든 언어들에서 다 정규식이 존재함 큰 틀에서는 거의 유사하고 언어별로 조금 차이는 있다
      • 파이썬 → re패키지(코테플랫폼: 기본으로 사용할 수 있는 패키지 중 하나)
      • 아나콘다 패키지 기본 설치, 코랩 기본 설치 → 불러다 쓰면 됨
    • 정규식으로 규칙을 나타내는 문자열 vs 쌩파이썬의 문자열
      • "hello.100$" --> 일반적인 문자열,,, r"hello.100$"
        --> 일반적인 문자열 앞에 r
        ⇒ 일반 문자열이 아니라,,,,정규식으로 작성된 문자열......
      • ()\[]*+ 기호들은 일반적인 특수문자가 아니라...
        -> 정규식 패키지에 사전에 정의를 해 놓은 규칙을 표현하는 문자!!!!!!!
    • 정규식 기본 틀
      • * vs +
        : 원하는 패턴이 0번 이상 vs 원하는 패턴이 1번 이상 ⇒ 출현빈도 기준점 제시
      • [문자들]
        : 여러개들 중에서 뭐 하나라도 걸렸으면 좋겠다…⇒ [Pp]ython
      • .
        : 줄바꿈을 제외하고 뭐라도 1개(문자일수도, 공백일수도, 숫자일수도…)를 얘기해줄 때(뭐가 올지 모를 때)
      • {n}
        : 패턴의 반복 횟수를 명시적 지정
        #0 ~ 9 까지 숫자 중 하나 체크
        [0123456789]
        
        [0-9]
        
        \\d #편하기는 하지만 구체적으로 컨트롤 불가능 
        #0~9까지를 명확히 표현 구간 변경 불가
        
        #ex)주민번호 XXXXXX - [1234]XXXXXX
        #ex)주민번호 XXXXXX - [1-4]XXXXXX
        ​
      • 정규식 활용해서 특정 패턴 찾을 때 : findall re.findall(패턴 → 규칙(정규식룰), 어디서)
        • ex) 찾을 규칙 : 숫자 1개로 이루어진 것들은 다 찾아주세요!!!![0~9] 
        • ex) 찾을 규칙 : 숫자대상으로 찾고 싶은데,, → 숫자 1개 이상으로 연결된 숫자덩어리를 찾아줘!!!




            
        • [a-zA-Z] : 대문자 소문자 다 가능
          [A-z] : 대문자 소문자 다 가능
          [a-Z] : 오류 
  • 사전지식2) 숫자 앞에 있는 0을 어떻게 처리?
    • 정렬을 하기 위해 0을 제거한 숫자
    • 000023 → 23
      int("000023")
      #강제 형 변환 : 제로 페딩을 숫자화가 가능하다
      ​

  • 사전지식1/2를 temp에 적용해보자

[1,3,3,3,4,3,3,3,2,3,3,3,3,2,5,5].index(4)

"sdsasdaserqwe42rafadf".index("d")

"sdsasdaserqwe42rafadf".index("z") #오류 #있다는 전제하에 사용해야함

"sdsasdaserqwe42rafadf".index("a") # 3

"sdsasdaserqwe42rafadf"[:3] # a앞까지 긁어오기
temp.index( re.findall( r'\\d+', temp)[0] )
#12

temp[: temp.index( re.findall( r'\\d+', temp)[0] )] # ---> head 파트!!
#Ssssssdsdsds

  • 사전지식3) 문자열 → 대소문자 통일!!


전체 코드 작성

  • 방법1) 리스트를 사용해서 구현
    • → (head:소문자, number:숫자, 원본위치)
      import re
      def solution(files):
          answer = []
      
      		#------------------------------------------------------
          #0. 기본 세팅 : 정렬에 필요한 정보들을 담을 변수 --> 리스트
          #               (head: 소문자, number: 숫자, 원본위치)
          my_files = [] #[(h,n,i), (h,n,i), .....]
      
      #for문 시작-----------------------------------------------------------------
      		#1. 큰 틀 : 입력으로 받은 모든 파일들을 롤링하면서 필요한 정보 추출
          #          + 위의 my_filesd에 담아두기
          #          + 원본 파일위치: 정수 인덱스
          for idx, s in enumerate(files):
      				#idx : 입력 리스트의 정수 인덱스,
      		    #s   : 개별 파일 명 --> h, n
      		    
      		    #1-1) 기준: 가장 처음에 나타나는 숫자 덩어리 ==> 정규식
              numbers = re.findall(r'\\d+', s) #foo010bar020.zip -> ['010','020']
      				#이 중에서 무조건 1개 이상은 존재하게 되니 정수인덱스
      		    #=> 첫번째 숫자 덩어리 : --> 완벽한 숫자화
              real_number = int(numbers[0]) # '010' --> int --> 10
      
      				#1-2) head 파트 처리 : 처음~~처음 숫자파트 시작 전까지
              head = s[:s.index(numbers[0])] # --> foo//if FOO(대문자라면?)
              head = head.lower() #FOO -> foo
      
      				#1-3) 추출하거나 처리할 것은 끝 -> 필요한 정보만 모아서 담자
      		    # 'foo', 10, 0 ==> ["foo", 10, 0]
      		    # "img", 3, 1 ==> ["img", 3, 1 ]
      		    #==> [["foo", 10, 0], ["img", 3, 1 ]]
              my_files.append([head, real_number, idx])
      #--> 이 for문이 끝나면 입력으로 받은 files에 대한 정렬을 위한 정보 처리 끝------------
      
      		#2) 문제에서 주어진 조건대로 정렬
      	  # + 일반적인 정렬 조건이 아니라 주어진대로 해야하는 경우, 명시적으로 작성해야함
      	  #.sort()/sorted()/ + key lambda + (+/-)
          my_files.sort(key = lambda x : (x[0],x[1],x[2]))
      		# x : [h, n, idx]
      	  # ex) [["xoo", 10, 0], ["img", 3, 1 ]] --> [["img", 3, 1 ], ["xoo", 10, 0]]
      	  
      	  #3) 최종 제출 스타일로 정리 --> 원본 파일명 값, 양식은 리스트로
          answer = [files[j[2]] for j in my_files]
      
          return answer
      ​
      #코드만 정리
      import re
      def solution(files):
          answer = []
          my_files = [] # [ (h,n,i), (h,n,i),,,,]
          for idx, s in enumerate(files):
              numbers = re.findall( r"\\d+", s) # foo010bar020.zip	 --> ["010","020"]
              real_number = int( numbers[0]) # "010" --> int --> 10
              head = s[:s.index( numbers[0] )] # --> foo // FOO
              head = head.lower() # Foo ---> lower ---> foo
              my_files.append( [head,real_number, idx] )
          my_files.sort(  key = lambda x: (x[0], x[1] , x[2]))
          answer = [ files[j[2]] for j in my_files]# if: (h,n,i,org) : j[3]
          return answer
      

  • 방법2) dict를 사용해서 구현
    • key
      • 문제 조건에서 의미상 동일한 것들 있을 수 있지만
      • 완벽하게 모양상 일치하는 파일명은 없음
      • 유니크함
    • value
      • 정렬에 필요한 정보 (h, n, org_idx)
        # Try1 ) 리스트 --> ( h, n, idx)
        import re
        def solution(files):
            answer = []
            # 세팅
            #my_files = [] # [ (h,n,i), (h,n,i),,,,]
            my_files = {} # k:파일명, v :(h,n,i)
            
            for idx, s in enumerate(files):
            #["img12.png", "img10.png"]라면
            #첫 번째는 s="img12.png", idx=0
            #두 번째는 s="img10.png", idx=1
            
                numbers = re.findall( r"\\d+", s)
                real_number = int( numbers[0])
                head = s[:s.index( numbers[0] )].lower()
                my_files[s] = (head, real_number, idx)
                # my_files.append( [head,real_number, idx] )
                # ==> {"Aa010.jpg":("aa", 10, 0), "aaaA01.png":("aaaa", 1, 1),,,}
            #
            file_list = my_files.items() #딕셔너리를 (키, 값) 쌍으로 바꿔요
            #{"Aa010.jpg":("aa", 10, 0), "aaaA01.png":("aaaa", 1, 1)} 에서 
            # => 이렇게 [("Aa010.jpg", ("aa", 10, 0) ), ("aaaA01.png",("aaaa", 1, 1) ) ]
            
            #my_files.sort(  key = lambda x: (x[0], x[1] , x[2]))
            file_list = sorted( file_list, key = lambda x: (x[1][0], x[1][1], x[1][2]) )
            #file_list = [("img12.png", ("img", 12, 0)),("img10.png", ("img", 10, 1)),("img2.png",  ("img", 2, 2))]
            #-> x[1][0] : Head, x[1][1] : number, x[1][2] : index
            
            #answer = [ files[j[2]] for j in my_files]# if: (h,n,i,org) : j[3]
            answer = [i[0] for i in file_list]
            return answer
        ​
        import re
        def solution(files):
            answer = []
            my_files = {} # k:파일명, v :(h,n,i)
            
            for idx, s in enumerate(files):
                numbers = re.findall( r"\\d+", s)
                real_number = int( numbers[0])
                head = s[:s.index( numbers[0] )].lower()
                my_files[s] = (head, real_number, idx)
             
            file_list = my_files.items()
            file_list = sorted( file_list, key = lambda x: (x[1][0], x[1][1], x[1][2]) )
            
            answer = [i[0] for i in file_list]
            
            return answer
        ​