[백준/BOJ] 3003번: 킹, 퀸, 룩, 비숍, 나이트, 폰 (Python 파이썬)

문제

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

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

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

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

 

입출력 규칙

1. 입력

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

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

 

---

문제 풀이

<내 풀이1>

 

for 문을 이용해서, 각 원소들을 하나씩 차이를 계산해 출력하였다.

total = [1,1,2,2,2,8]

l = list(map(int,input().split()))
for i in range(len(total)):
    total[i] = total[i] - l[i]
print(*total)

 

<숏코딩>

 

1. 위에 내가 짠 코드를 한줄로 짠것이다.    변수에 할당하지 않고 바로 계산하여 출력해주는 코드이다.

print(b-int(a) for a,b in zip(input().split(),[1,1,2,2,2,8]))

 

2. eval() 함수와 join() 함수를 너무 잘 활용한 예제이다.

print(*map(eval,map('-'.join,zip('112228',input().split()))))

 

 

1. zip('112228',input().split()) : '112228'과 입력받은 문자열을 묶는 것이다.

   ex. 입력한 문자열이 '1 2 3 4 5 6'이라면,  ('1', '1') ('1', '2') ('2', '3') ('2', '4') ('2', '5') ('8', '6') 처럼 묶인다.

      

2. map('-'.join,zip('112228',input().split())) : map 함수는 일괄처리하는 것으로, zip한 결과물을 '-'으로 결합하는 것이다.

    ex.  ('1', '1') ('1', '2') ('2', '3') ('2', '4') ('2', '5') ('8', '6') 이였던 것을 '-'로 결합하면, 1-1   1-2   2-3   2-4   2-5   8-6

 

3. map(eval,map('-'.join,zip('112228',input().split()))) : eval() 함수로 문자열 식을 일괄적으로 계산하면 된다.

   ex.  1-1   1-2   2-3   2-4   2-5   8-6 이였던 것을 eval함수로 계산하면, 0 -1 -1 -2 -3 2

 

4. unpakcing(*)을 통해서 결과물을 출력하면 된다.

 

+ 참고

zip() 함수: 여러 개의 iterable 객체를 인자로 받고, 각 객체가 담고 있는 원소를 튜플 형태로 차례로 접근할 수 있는 iterator를 반환한다.

a = [1, 2, 3]
b = ['a', 'b','c']

for pair in zip(a,b):
    print(pair) # (1, 'a')
                # (2, 'b')
                # (3, 'c')