Python/알고리즘

[파이썬 알고리즘 인터뷰] 3. 로그 파일 재정렬

AI 꿈나무 2021. 2. 9. 14:45
반응형

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 


3. 로그 파일 재정렬

리트코드 937번 문제입니다.

 

풀이

 sort함수 key인자에 lambda를 활용하여 [1] 인덱스 기준으로 정렬하고 값이 동일하면 [0] 인덱스로 정렬하도록 하는 것이 핵심이었습니다.

def reorderLogFiles(self, logs):
    digits, letters = [], []

    for log in logs:
        if log[1].isdigit():
            digits.append(log)
        else:
            letters.append(log)

    letters.sort(key=lambda x: (x.split()[1], x.split()[0]))
    return letters + digits
반응형