[백준] 3003번 - 킹, 퀸, 룩, 비숍, 나이트, 폰

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

 

✔️ 문제

 

 

문제는 다음과 같다.

 

 

✔️ 풀이

 

 

예제를 보게 되면 한 가지 공통점을 발견할 수 있다.

기존 수량에서 들어온 입력 값을 빼주면,

출력에 나오는 값과 일치하는 것이다.

 

import sys
d=list(map(int,sys.stdin.readline().rstrip().split()))
p=[1,1,2,2,2,8]
for i in range(len(d)):
    if d[i]>p[i] or d[i]<p[i]:
        print(p[i]-d[i], end=' ')
    else:
        print(0, end=' ')

 

처음 코드를 짤 때는 다음과 같이 생각했지만,

굳이 if문을 사용하지 않고도 풀 수 있다는 생각이 들어

다시 한번 코드를 정리했다.

 

import sys
d=list(map(int,sys.stdin.readline().rstrip().split()))
p=[1,1,2,2,2,8]
for i in range(len(d)):
    print(p[i]-d[i], end=' ')

 

기존 갯수를 미리 만들어놓고 들어오는

입력 값을 빼주면 된다.

 

화이팅 💪