[BOJ#3003번] list(zip()), 두 리스트의 원소를 순서대로 짝을 맞춰 엮는 방법
* 킹, 퀸, 룩, 비숍, 나이트, 폰 (3003번)
문제
동혁이는 오래된 창고를 뒤지다가 낡은 체스판과 피스를 발견했다.
체스판의 먼지를 털어내고 걸레로 닦으니 그럭저럭 쓸만한 체스판이 되었다. 하지만, 검정색 피스는 모두 있었으나, 흰색 피스는 개수가 올바르지 않았다.
체스는 총 16개의 피스를 사용하며, 킹 1개, 퀸 1개, 룩 2개, 비숍 2개, 나이트 2개, 폰 8개로 구성되어 있다.
동혁이가 발견한 흰색 피스의 개수가 주어졌을 때, 몇 개를 더하거나 빼야 올바른 세트가 되는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 동혁이가 찾은 흰색 킹, 퀸, 룩, 비숍, 나이트, 폰의 개수가 주어진다. 이 값은 0보다 크거나 같고 10보다 작거나 같은 정수이다.
출력
첫째 줄에 입력에서 주어진 순서대로 몇 개의 피스를 더하거나 빼야 되는지를 출력한다. 만약 수가 양수라면 동혁이는 그 개수 만큼 피스를 더해야 하는 것이고, 음수라면 제거해야 하는 것이다.
print(*(e-f for e,f in zip((1,1,2,2,2,8),map(int,input().split()))))
* zip() 내장함수
Iterate over several iterables in parallel, producing tuples with an item from each one
여러 반복 가능한 항목을 병렬로 반복하여 각각의 항목이 포함된 튜플을 생성합니다
for item in zip([1, 2, 3], ["sugar", "spice", "everything nice"]):
print(item)
----------------------------------------------
(1, 'sugar')
(2, 'spice')
(3, 'everything nice')
zip()함수를 for문으로 바꿔 표현하면 아래와 같다.
numbers = [1, 2, 3]
letters = ["sugar", "spice", "everything nice"]
for i in range(3):
pair = (numbers[i], letters[i])
print(pair)
---------------------------------------
(1, 'sugar')
(2, 'spice')
(3, 'everything nice')
3개 이상의 인자도 병렬 처리 가능하다.
for number, upper, lower in zip("12345", "ABCDE", "abcde"):
print(number, upper, lower)
for item in zip("12345", "ABCDE", "abcde"):
print(item)
--------------------------------------------
1 A a
2 B b
3 C c
4 D d
5 E e
('1', 'A', 'a')
('2', 'B', 'b')
('3', 'C', 'c')
('4', 'D', 'd')
('5', 'E', 'e')
* unzip 방법
1. list로 변환
zip() 함수에 넣는 인자의 길이가 다르면 가장 짧은 인자를 기준으로 엮이고, 나머지는 버려진다.
arr = list(zip(range(3), ["fee", "fi", "fo", "fum"]))
print(arr)
-----------------------------------------------
[(0, 'fee'), (1, 'fi'), (2, 'fo')]
길이가 다른 인자가 zip()함수에 의해 엮이는 것을 방지하고자 할 때는 strict=True 옵션을 적용한다.
for item in zip(range(3), ["fee", "fi", "fo", "fum"], strict=True):
print(item)
---------------------------------------------------------------------
Exception has occurred: ValueError
...
ValueError: zip() argument 2 is longer than argument 1
2. *연산자를 사용한 unzip방법
x = [1, 2, 3]
y = [4, 5, 6]
x2, y2 = zip(*zip(x, y))
print(x2, y2)
-----------------------------
(1, 2, 3) (4, 5, 6)
3. Dictionary로 변환
keys = [1, 2, 3]
values = ["A", "B", "C"]
print(dict(zip(keys, values)))
------------------------------
{1: 'A', 2: 'B', 3: 'C'}
* 참고
https://docs.python.org/ko/3/library/functions.html?highlight=zip#zip