4.py

# 2차원 리스트 90도 회전하기
defrotate_a_matrix_by_90_degree(a):
n=len(a) # 행 길이 계산
m=len(a[0]) # 열 길이 계산
result= [[0] *nfor_inrange(m)] # 결과 리스트
foriinrange(n):
forjinrange(m):
result[j][n-i-1] =a[i][j]
returnresult

# 자물쇠의 중간 부분이 모두 1인지 확인
defcheck(new_lock):
lock_length=len(new_lock) //3
foriinrange(lock_length, lock_length*2):
forjinrange(lock_length, lock_length*2):
ifnew_lock[i][j] !=1:
returnFalse
returnTrue

defsolution(key, lock):
n=len(lock)
m=len(key)
# 자물쇠의 크기를 기존의 3배로 변환
new_lock= [[0] * (n*3) for_inrange(n*3)]
# 새로운 자물쇠의 중앙 부분에 기존의 자물쇠 넣기
foriinrange(n):
forjinrange(n):
new_lock[i+n][j+n] =lock[i][j]

# 4가지 방향에 대해서 확인
forrotationinrange(4):
key=rotate_a_matrix_by_90_degree(key) # 열쇠 회전
forxinrange(n*2):
foryinrange(n*2):
# 자물쇠에 열쇠를 끼워 넣기
foriinrange(m):
forjinrange(m):
new_lock[x+i][y+j] +=key[i][j]
# 새로운 자물쇠에 열쇠가 정확히 들어 맞는지 검사
ifcheck(new_lock) ==True:
returnTrue
# 자물쇠에서 열쇠를 다시 빼기
foriinrange(m):
forjinrange(m):
new_lock[x+i][y+j] -=key[i][j]
returnFalse