Python/백준 알고리즘
[백준][Python3] #3003. 킹, 퀸, 룩, 비숍, 나이트, 폰
2nan
2023. 1. 14. 18:04
728x90
코드 입력
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 함수를 많이 사용하였는데, 해당 함수의 역할이나 느낌을 아직 정확히 알지 못 해서 다시 정리하기로 했다. O Map의 정의 - map은 리스트의 요소를 지정된 함수로
2nan.tistory.com
어쨌든, map 함수에 split 을 이용해서 여러 개를 한꺼번에 input으로 받아서 처리하는 로직을 사용해서 비교적 간단하게 해결한 것 같다!
출처 : 백준 Online Judge
https://www.acmicpc.net/problem/3003
3003번: 킹, 퀸, 룩, 비숍, 나이트, 폰
첫째 줄에 동혁이가 찾은 흰색 킹, 퀸, 룩, 비숍, 나이트, 폰의 개수가 주어진다. 이 값은 0보다 크거나 같고 10보다 작거나 같은 정수이다.
www.acmicpc.net