알고리즘 문제풀이

시리얼 번호(백준 1431)

wiojfe 2024. 3. 5. 17:34

 

https://www.acmicpc.net/problem/1431

 

1431번: 시리얼 번호

첫째 줄에 기타의 개수 N이 주어진다. N은 50보다 작거나 같다. 둘째 줄부터 N개의 줄에 시리얼 번호가 하나씩 주어진다. 시리얼 번호의 길이는 최대 50이고, 알파벳 대문자 또는 숫자로만 이루어

www.acmicpc.net

 문제의 조건에 맞게 정렬을 람다 함수를 활용하여 해준다. 

n = int(input())
def num_sum(string):
    ans =0 
    for i in range(len(string)):
        if ord(string[i])>=65:continue 
        else :ans+= int(string[i])
    return ans 
        
    
arr = []
for i in range(n):
    string =input().rstrip()
    arr.append((string, num_sum(string)))

arr.sort(key =lambda x :(len(x[0]),x[1], x[0]))
for i in range(n):
    print(arr[i][0])