[백준/Python] 5576 - 콘테스트

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

초기 코드

w = []
k = []
for _ in range(10):
    a = int(input())
    w.append(a)

for _ in range(10):
    a = int(input())
    k.append(a)

w.sort()
k.sort()

print(sum(w[-3:]), sum(k[-3:]))

코드 개선하기

입력받는 부분을 더 간단하게 만들 수 있다.

w = [int(input()) for _ in range(10)]
k = [int(input()) for _ in range(10)]

w.sort()
k.sort()

print(sum(w[-3:]), sum(k[-3:]))