[백준] 3003번 파이썬(Python)

문제

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

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

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

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

입력

첫째 줄에 동혁이가 찾은 흰색 킹, 퀸, 룩, 비숍, 나이트, 폰의 개수가 주어진다. 이 값은 0보다 크거나 같고 10보다 작거나 같은 정수이다.

출력

첫째 줄에 입력에서 주어진 순서대로 몇 개의 피스를 더하거나 빼야 되는지를 출력한다. 만약 수가 양수라면 동혁이는 그 개수 만큼 피스를 더해야 하는 것이고, 음수라면 제거해야 하는 것이다.

---

'정답 코드'

chess = list(map(int, input().split()))

for i in range(len(chess)):
    # 킹, 퀸의 경우
    if i == 0 or i == 1:
        if chess[i] == 0:
            chess[i] = 1
        elif chess[i] == 1:
            chess[i] = 0
        else:
            chess[i] = 1 - chess[i]
            
    # 룩, 비숍, 나이트의 경우    
    if i == 2 or i == 3 or i == 4:
        if chess[i] < 2:
            chess[i] = 2 - chess[i]
        elif chess[i] == 2:
            chess[i] = 0
        else:
            chess[i] = 2 - chess[i]
            
    if i == 5:
        if chess[i] < 8:
            chess[i] = 8 - chess[i]
        elif chess[i] == 8:
            chess[i] = 0
        else:
            chess[i] = 8 - chess[i]
        
print(*chess)

개수가 같은 피스들끼리 묶어 풀었다.

 

'돌아보며'

처음에 print(' '.join(chess))로 구현하여 런타임 에러가 났다.

찾아보니 join할 때는 string형이 들어가야 하는데 내가 구현한 chess리스트의 원소는 int형이었기 때문이다.

만약, 숫자형 리스트를 join 하고 싶다면 어떻게 할까?

map() 메소드를 사용해 str형으로 형변환 해주면 끝!

print(' '.join(map(str, chess)))