[python]boj1431

[chjcode]

📣 Related Issue

close #

🌁 Background

파이썬 정렬문제에서 cmp_to_key()를 이용하는 방법 공유
cmp_to_key()는 1,0,-1 세 가지 값을 받는데, 1을 받을 경우 두 수의 자리를 바꾸고, 0과 -1은 자리를 바꾸지 않는다고 생각하면 훨씬 간단해진다.

👩‍💻 Contents

# 1431 시리얼 번호

n = int(input())
serial = [input() for _ in range(n)]

from functools import cmp_to_key

def compare(a,b):
    len_a = len(a)
    len_b = len(b)
    num = ['0','1','2','3','4','5','6','7','8','9']
    # 첫번째 조건
    if len_a > len_b:   # a가 큰데 자리를 바꾸면 큰 값이 뒤로 오게 정렬됨
        return 1
    elif len_a < len_b:
        return -1

    # 두번째 조건
    cnt_a = 0
    cnt_b = 0
    for i in range(len_a):
        if a[i] in num:
            cnt_a += int(a[i])
        if b[i] in num:
            cnt_b += int(b[i])
    if cnt_a > cnt_b:
        return 1
    elif cnt_a < cnt_b:
        return -1
    # 세번째 조건
    if a > b:
        return 1
    else:
        return -1


serial = sorted(serial, key = cmp_to_key(compare))

for temp in serial:
    print(temp)

📱 Screenshot

📝 Review Note

📬 Reference

[cJinu]

잘 보았습니다!