33-Dolchae

[Dolchae]

🔗 문제 링크

1431 시리얼 번호

✔️ 소요된 시간

1시간

✨ 수도 코드

문제

시리얼 번호를 정렬하는 프로그램을 구현하는 문제이다. 정렬은 다음과 같이 이루어진다.

1. 길이는 짧은 것이 먼저(앞에) 온다.
2. 길이가 같다면 각 모든 자릿수의 합이 작은 것이 먼저 온다.(숫자인 것만 더함)
3. 1,2로도 비교할 수 없다면 사전순으로 비교한다. 이때 숫자가 알파벳보다 사전순으로 더 앞이다.

코드

import sys
input = sys.stdin.readline

n = int(input().strip())
serial = []

for _ in range(n):
    serial.append(input().strip())

n에 시리얼 번호의 개수, serial 리스트에 시리얼 번호를 모두 저장한다.

def get_number_sum(s):
    number_sum = 0
    for char in s:
        if char.isdigit():
            number_sum += int(char)
    return number_sum

후의 정렬에서 각 번호의 숫자합을 구할 때 쓸 함수이다. 문자열 s를 받아 문자 char가 정수라면 그 값을 number_sum에 저장한 후 이 값을 넘긴다.

for i in range(n-1):
    for j in range(i+1, n):
        if len(serial[i]) > len(serial[j]): #1
            serial[j], serial[i] = serial[i], serial[j]
        elif len(serial[i]) == len(serial[j]): #2
            i_total = get_number_sum(serial[i])
            j_total = get_number_sum(serial[j])
            if i_total > j_total: #3
                serial[j], serial[i] = serial[i], serial[j]
            elif i_total == j_total: #4
                if serial[i] > serial[j]:
                    serial[j], serial[i] = serial[i], serial[j]

for i in serial:
    print(i)

for문 i는 (0 ~ n-2), j는 (i+1 ~ n-1)까지 반복하며 정렬 과정을 수행한다.
if문 #1에서 만약 i가 j보다 길다면 둘의 위치를 바꾸어준다.
elif문 #2에서 i와 j의 길이가 같다면 숫자 합을 더해줘야하므로 get_number_sum을 호출해 숫자합을 i_total, j_total에 각각 저장한다.
if문 #3에서 i_total보다 j_total이 더 작다면 둘의 위치를 바꾸어준다.
elif문 #4에서 i_total, j_total이 같다면 알파벳 순까지 신경써야하므로 j가 더 알파벳순으로 앞이라면 둘의 위치를 바꾸어준다.

마지막으로 serial 리스트를 출력해준다.

전체 코드

import sys
input = sys.stdin.readline

n = int(input().strip())
serial = []

for _ in range(n):
    serial.append(input().strip())

def get_number_sum(s):
    number_sum = 0
    for char in s:
        if char.isdigit():
            number_sum += int(char)
    return number_sum

for i in range(n-1):
    for j in range(i+1, n):
        if len(serial[i]) > len(serial[j]):
            serial[j], serial[i] = serial[i], serial[j]
        elif len(serial[i]) == len(serial[j]):
            i_total = get_number_sum(serial[i])
            j_total = get_number_sum(serial[j])
            if i_total > j_total:
                serial[j], serial[i] = serial[i], serial[j]
            elif i_total == j_total:
                if serial[i] > serial[j]:
                    serial[j], serial[i] = serial[i], serial[j]

for i in serial:
    print(i)

📚 새롭게 알게된 내용

아직도 내 눈에는 문제점이 보이지 않는데🥹 밑의 코드가 자꾸 j for문에서 "TypeError: 'str' object cannot be interpreted as an integer"에러가 떠서 GPT의 도움을 받았다. 그리고 다른 분들 코드를 보니 내장함수를 사용하면 더 빨리 처리할 수 있는 것 같다..!

[9kyo-hwang]

파이썬 코드는 '파이써닉'하게 짜는 것 또한 중요한 능력입니다.

C++이나 Java같은 다른 언어를 쓰던 사람이 파이썬을 잡으면, 이전에 작성하던 버릇이 남아있어 그 언어를 쓰던 방식대로 코드를 작성하곤 합니다. 그러면 "파이써닉하지 못하다"라고 얘기합니다. 파이썬의 장점인 간결하고 내부적으로 최적화된 코드를 사용하지 못하기 때문이죠.

한 번 이 부분도 시간나면 검색해보시는 걸 추천합니다 :)

[9kyo-hwang]

늘 말 하던 것이지만,

serial = list(input().rstrip() for _ in range(n))

으로 줄여쓸 수 있겠죠?

[9kyo-hwang]

def get_number_sum(s):
    return sum(int(char) for char in s if char.isdigit())

요렇게 한 줄로 간단하게 줄일 수 있답니다 :)

[9kyo-hwang]

파이썬의 sort는 강력합니다. 정렬 기준인 key를 잘 정해주기만 하면 되니까요.

이 문제의 경우 1. 길이가 짧은 순 2. 자릿수 합이 적은 순 3. 사전순 우선순위로 정렬해야 하잖아요? 그러니까

serial.sort(key=lambda x: (len(x), get_number_sum(x), x))

이렇게 람다로 정해주면 됩니다. x가 각각의 원소인 문자열을 의미하므로

1. 길이가 짧은 순으로
2. get_number_sum 함수를 호출한 결과(자릿수 합)가 작은 순으로
3. 문자열 사전 순으로

정렬하게 됩니다 :)

[xxubin04]

chat GPT를 잘 활용하시는 모습이 멋지십니다!!👍
저는 항상 구글링으로만 해결하려 했는데, 본받을 수 있도록 하겠습니다!😊
그리고 저도 내장함수 잘 안쓰는데 쓰려고 노력해야겠습니다.
수고하셨습니다!!

[gjsk132]

문자열을 특정 조건으로 정렬하는 건 자료구조 강의 때에도 많이 했지만 여전히 적응이 안되네요 😭
중간에 이상하게 계속 줄바꿈이 하나씩 살펴보며 고치다가 한참 헤맸어요...
그래도 어찌저찌 풀었습니다..!

이번 PR도 수고하셨어요!!

코드

input_ = open(0).readline

from collections import defaultdict

words = [input_().strip() for _ in range(int(input_()))]

len_words = defaultdict(list)

for w in words:
    len_words[len(w)].append(w)

for same_len in sorted(len_words):
    cnt_words = defaultdict(list)
    for w in len_words[same_len]:
        w_sum = sum(int(p) for p in w if p.isdigit())
        cnt_words[w_sum].append(w)

    for i in sorted(cnt_words):
        for j in sorted(cnt_words[i]):
            print(j, end=("\n" if j else ""))