#include
using namespace std;
// 2차원 리스트 90도 회전하기
vector > rotateMatrixBy90Degree(vector > a) {
int n = a.size(); // 행 길이 계산
int m = a[0].size(); // 열 길이 계산
vector > result(n, vector(m)); // 결과 리스트
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
result[j][n - i - 1] = a[i][j];
}
}
return result;
}
// 자물쇠의 중간 부분이 모두 1인지 확인
bool check(vector > newLock) {
int lockLength = newLock.size() / 3;
for (int i = lockLength; i < lockLength * 2; i++) {
for (int j = lockLength; j < lockLength * 2; j++) {
if (newLock[i][j] != 1) {
return false;
}
}
}
return true;
}
bool solution(vector > key, vector > lock) {
int n = lock.size();
int m = key.size();
// 자물쇠의 크기를 기존의 3배로 변환
vector > newLock(n * 3, vector(n * 3));
// 새로운 자물쇠의 중앙 부분에 기존의 자물쇠 넣기
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
newLock[i + n][j + n] = lock[i][j];
}
}
// 4가지 방향에 대해서 확인
for (int rotation = 0; rotation < 4; rotation++) {
key = rotateMatrixBy90Degree(key); // 열쇠 회전
for (int x = 0; x < n * 2; x++) {
for (int y = 0; y < n * 2; y++) {
// 자물쇠에 열쇠를 끼워 넣기
for (int i = 0; i < m; i++) {
for (int j = 0; j < m; j++) {
newLock[x + i][y + j] += key[i][j];
}
}
// 새로운 자물쇠에 열쇠가 정확히 들어 맞는지 검사
if (check(newLock)) return true;
// 자물쇠에서 열쇠를 다시 빼기
for (int i = 0; i < m; i++) {
for (int j = 0; j < m; j++) {
newLock[x + i][y + j] -= key[i][j];
}
}
}
}
}
return false;
}