[백준 Python] 3003번 킹, 퀸, 룩, 비숍, 나이트, 폰 문제풀이

문제

동혁이는 오래된 창고를 뒤지다가 낡은 체스판과 피스를 발견했다.

체스판의 먼지를 털어내고 걸레로 닦으니 그럭저럭 쓸만한 체스판이 되었다. 하지만, 검정색 피스는 모두 있었으나, 흰색 피스는 개수가 올바르지 않았다.

체스는 총 16개의 피스를 사용하며, 킹 1개, 퀸 1개, 룩 2개, 비숍 2개, 나이트 2개, 폰 8개로 구성되어 있다.

동혁이가 발견한 흰색 피스의 개수가 주어졌을 때, 몇 개를 더하거나 빼야 올바른 세트가 되는지 구하는 프로그램을 작성하시오.

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

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

 

풀이

index = 0
correct_piece = [1, 1, 2, 2, 2, 8] # 올바른 체스 숫자
numbers = list(map(int, input().split())) # 입력받는 숫자
answer_list = []  # 정답

for number in numbers:
    if number == correct_piece[index]: # 만약 두 숫자가 같다면 0
        answer_list.append(0)
    else: # 그렇지 않다면 뺀 값을 리스트에 넣는다. 음수도 필요하기 때문에 값을 이렇게 넣기
        answer_list.append(correct_piece[index]-number)
        
    index += 1

for answer in answer_list:
    print(answer, end=' ')