코딩테스트
[Programmers][2020 KAKAO BLIND RECRUITMENT] 자물쇠와 열쇠
hoon123
2023. 3. 21. 22:56
혼자서 열심히 풀려고 했지만 1시간 안에 푸는 게 불가능해 답을 보면서 공부하기로 했다. ㅠ
풀이 힌트
1. key 회전
2. 자물쇠 확장
import copy
def expend_lock(lock, N, M, size):
# 자물쇠를 확장시키는 함수
# 순수 함수
expended_lock = [[0 for i in range(size)] for _ in range(size)]
for y in range(N):
for x in range(N):
expended_lock[y + M - 1][x + M - 1] = lock[y][x]
return expended_lock
def rotate(key):
# *key는 껍질(Unpacking)을 벗겨내는 역할을 합니다. [] () {}
# [(0, 0, 0), (1, 0, 0), (0, 1, 1)] *key
# [(0, 1, 0), (0, 0, 1), (0, 0, 1)] zip(*key)
# [(0, 1, 0), (1, 0, 0), (1, 0, 0)] reversed
# [[0, 1, 0]] ... -> list
# 순수 함수
return [list(reversed(i)) for i in zip(*key)]
def is_unlock(y, x, key, lock, N, M):
# 순수 함수
# 순수 함수가 아니면 Side Effect가 있을 수 있는 함수라고 한다.
_lock = copy.deepcopy(lock)
for _y in range(M):
for _x in range(M):
_lock[_y + y][_x + x] += key[_y][_x]
for _y in range(N):
for _x in range(N):
if _lock[_y + M - 1][_x + M - 1] != 1:
return False
return True
def solution(key, lock):
N = len(lock)
M = len(key)
size = (N - 1) * 2 + M # 최소 1개가 겹쳐야되니 N-1
expended_lock = expend_lock(lock, N, M, size)
for _ in range(4):
for y in range(size - M + 1):
for x in range(size - M + 1):
if is_unlock(y, x, key, expended_lock, N, M):
return True
key = rotate(key)
return False