코드 입력
king = 1
queen = 1
rook = 2
bishop = 2
knight = 2
pawn = 8
def get_diff(n, chess):
if n != chess:
result = chess - n
return result
else:
return 0
def chess(k, q, r, b, kn, p):
k_ = get_diff(k, king)
q_ = get_diff(q, queen)
r_ = get_diff(r, rook)
b_ = get_diff(b, bishop)
kn_ = get_diff(kn, knight)
p_ = get_diff(p, pawn)
return k_, q_, r_, b_, kn_, p_
a,b,c,d,e,f = map(int, input().split())
a_, b_, c_, d_, e_, f_ = chess(a, b, c, d, e, f)
생각보다 간단한 문제인데 표현하는데서 애먹은 문제다.
기준 값하고 다를 경우에 차이 값을 return 해주고, 같은 경우에는 0만 리턴해주면 되는 문제였다.
근데, input 값을 여러 개 표현하는데 어떻게 했었지..? 하면서 구글링을 하다가 map을 다시 떠올리게 했다.
분명 따로 정리도 했던 것 같은데, 이후로 쓰질 않아서 기억에서 삭제되었나보다.
어쨌든, map 함수에 split 을 이용해서 여러 개를 한꺼번에 input으로 받아서 처리하는 로직을 사용해서 비교적 간단하게 해결한 것 같다!