[python] 백준 3003번 풀이

내 정답

킹 1개, 퀸 1개, 룩 2개, 비숍 2개, 나이트 2개, 폰 8개
plus_mi = list(map(int, input().split()))
sol = [1, 1, 2, 2, 2, 8]
fix = [0 for i in range(6)]
for i in range(6):
    if plus_mi[i] != sol[i]:
        fix[i] = sol[i] - plus_mi[i]
for i in range(6):
    print(fix[i], end=' ')        

보완할 점

plus_mi = list(map(int, input().split()))
sol = [1, 1, 2, 2, 2, 8]
for i in range(6):
    print(sol[i] - plus_mi[i], end=' ')           

굳이 if문을 쓸 필요가 없는게, 같아서 빼도 그냥 0이 출력된다.