[백준][Python3] #3003. 킹, 퀸, 룩, 비숍, 나이트, 폰

---

코드 입력

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)

print(a_, b_, c_, d_, e_, f_)

 

생각보다 간단한 문제인데 표현하는데서 애먹은 문제다.

기준 값하고 다를 경우에 차이 값을 return 해주고, 같은 경우에는 0만 리턴해주면 되는 문제였다.

근데, input 값을 여러 개 표현하는데 어떻게 했었지..? 하면서 구글링을 하다가 map을 다시 떠올리게 했다.

분명 따로 정리도 했던 것 같은데, 이후로 쓰질 않아서 기억에서 삭제되었나보다.https://2nan.tistory.com/48

[Python3][복습정리]#0812_map

 

어쨌든, map 함수에 split 을 이용해서 여러 개를 한꺼번에 input으로 받아서 처리하는 로직을 사용해서 비교적 간단하게 해결한 것 같다!

 

---

출처 : 백준 Online Judge

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

3003번: 킹, 퀸, 룩, 비숍, 나이트, 폰