티스토리 뷰
알고리즘
[알고리즘] 자물쇠와 열쇠_프로그래머스 레벨3_2차원 배열(array), 행렬(matrix), 전치 행렬(transpose)
데브캣_DevCat 2023. 12. 31. 23:14문제링크
📝 문제
고고학자인 "튜브"는 고대 유적지에서 보물과 유적이 가득할 것으로 추정되는 비밀의 문을 발견하였습니다. 그런데 문을 열려고 살펴보니 특이한 형태의 자물쇠로 잠겨 있었고 문 앞에는 특이한 형태의 열쇠와 함께 자물쇠를 푸는 방법에 대해 다음과 같이 설명해 주는 종이가 발견되었습니다.
잠겨있는 자물쇠는 격자 한 칸의 크기가 1 x 1
인 N x N
크기의 정사각 격자 형태이고 특이한 모양의 열쇠는 M x M
크기인 정사각 격자 형태로 되어 있습니다.
자물쇠에는 홈이 파여 있고 열쇠 또한 홈과 돌기 부분이 있습니다. 열쇠는 회전과 이동이 가능하며 열쇠의 돌기 부분을 자물쇠의 홈 부분에 딱 맞게 채우면 자물쇠가 열리게 되는 구조입니다. 자물쇠 영역을 벗어난 부분에 있는 열쇠의 홈과 돌기는 자물쇠를 여는 데 영향을 주지 않지만, 자물쇠 영역 내에서는 열쇠의 돌기 부분과 자물쇠의 홈 부분이 정확히 일치해야 하며 열쇠의 돌기와 자물쇠의 돌기가 만나서는 안됩니다. 또한 자물쇠의 모든 홈을 채워 비어있는 곳이 없어야 자물쇠를 열 수 있습니다.
열쇠를 나타내는 2차원 배열 key와 자물쇠를 나타내는 2차원 배열 lock이 매개변수로 주어질 때, 열쇠로 자물쇠를 열수 있으면 true를, 열 수 없으면 false를 return 하도록 solution 함수를 완성해주세요.
제한사항
- key는 M x M(3 ≤ M ≤ 20, M은 자연수)크기 2차원 배열입니다.
- lock은 N x N(3 ≤ N ≤ 20, N은 자연수)크기 2차원 배열입니다.
- M은 항상 N 이하입니다.
- key와 lock의 원소는 0 또는 1로 이루어져 있습니다.
- 0은 홈 부분, 1은 돌기 부분을 나타냅니다.
입출력 예
key | lock | result |
---|---|---|
[[0, 0, 0], [1, 0, 0], [0, 1, 1]] | [[1, 1, 1], [1, 1, 0], [1, 0, 1]] | true |
입출력 예에 대한 설명
key를 시계 방향으로 90도 회전하고, 오른쪽으로 한 칸, 아래로 한 칸 이동하면 lock의 홈 부분을 정확히 모두 채울 수 있습니다.
💡 풀이
- 주어진 키로 자물쇠(Lock)의 홈을 모두 맞출 수 있는지 여부를 알아내야 하는 문제입니다. 주요 조건으로는 키가 자물쇠 외부에도 위치할 수 있다는 것과 키를 90도씩 회전시킬 수 있다는 것입니다.
- 키와 자물쇠의 일치 여부는 배열을 확장시켜서 해결할 수 있습니다.
- 위와 같이 자물쇠 외부까지 설정한 배열을 탐색하는 것이죠! 이 때 각 행과 열의 길이는 자물쇠의 길이와 키의 길이 * 2 - 2 를 해주게 됩니다. 키가 왼쪽에서 시작해서 오른쪽 끝까지 갈 수 있어야 하므로 * 2 를 해주되 양 끝을 하나씩 포함하는 구조이기 때문입니다.
int k = key.length;
int l = lock.length;
int size = l + k * 2 - 2;
map = new int[size][size];
- map 배열에서 자물쇠 외부는 탐색할 필요가 없으므로 자물쇠 구역인지를 확인하는 배열(boolean isLockSection)을 하나 더 선언하였고 자물쇠의 홈 부분을 모두 키의 돌기로 채워야 하기 때문에 자물쇠의 홈 개수(int lockCount) 도 미리 세어두었습니다.
- 그리고 map 의 첫 부분부터 자물쇠 부분만 탐색을 하는 것이죠!
- map의 해당 위치에서 매 탐색마다 탐색의 범위는 key의 크기(녹색 부분)이고 맵의 인덱스와 키의 인덱스를 대응시켜 주기 위해서 맵의 시작점을 넘겨주어서 탐색하였습니다.
- 위 그림을 예로 들면 시작 위치 row, col = (1, 1) 을 넘겨주어서 map의 1, 1과 key의 0, 0 을 시작으로 비교하는 것이죠!
public boolean move(int[][] key, int[][] lock, int lockCount) {
for (int i = 0; i < key.length + lock.length - 1; i++) {
for (int j = 0; j < key.length + lock.length - 1; j++) {
if (checking(key, i, j, lockCount)) return true;
}
}
return false;
}
public boolean checking(int[][] key, int row, int col, int lockCount) {
int count = lockCount;
for (int i = 0; i < key.length; i++) {
for (int j = 0; j < key[0].length; j++) {
if (isLockSection[i+row][j+col]) {
if (map[i+row][j+col] == key[i][j]) return false;
else if (map[i+row][j+col] == 0 && key[i][j] == 1) {
count--;
}
}
}
}
return count == 0;
}
- 이 때 자물쇠 외부는 확인할 필요가 없기 때문에 lockSection = true 인 곳만, 그리고 홈과 홈이 만나거나 돌기와 돌기가 만난다면 false를 리턴하도록 하고 그 외 키의 돌기와 자물쇠의 홈이 자물쇠의 홈 개수만큼 만났을 때 true를 리턴하도록 하였습니다.
- 그리고 키는 90도씩 회전시킬 수 있기 때문에 이러한 탐색은 총 4번 이루어져야 하는데요! 이 알고리즘은 전치행렬을 이용해서 구현하였습니다.
- 먼저 전치행렬(transpose) 을 만들고 양 끝 컬럼 교환들을 해주면 되는 것인데요!
- 전치행렬이란 어떠한 행렬의 열과 행을 바꾸어낸 행렬을 의미합니다.
- 예를 들어서 [1, 2, 3], [4, 5, 6] 을 전치하면, [1, 4], [2, 5], [3, 6] 으로 3 x 2 행렬이 2 x 3 이 되는 것이죠!
- 만약 문제에서처럼 행과 열이 같다면 위와 같이 대각선을 기준으로 스왑이 이루어집니다.
🔍 정답
import java.util.*;
class Solution {
static int[][] map;
static boolean[][] isLockSection;
public boolean solution(int[][] key, int[][] lock) {
int k = key.length;
int l = lock.length;
int size = l + k * 2 - 2;
map = new int[size][size];
isLockSection = new boolean[size][size];
int rowStart = key.length - 1;
int colStart = key[0].length - 1;
int lockCount = 0;
for (int i = 0; i < lock.length; i++) {
for (int j = 0; j < lock[0].length; j++) {
if (lock[i][j] == 0) lockCount++;
map[rowStart+i][colStart+j] = lock[i][j];
isLockSection[rowStart+i][colStart+j] = true;
}
}
for (int i = 0; i < 4; i++) {
if (move(key, lock,lockCount)) return true;
rotate(key);
}
return false;
}
public void rotate(int[][] key) {
for (int i = 0; i < key.length-1; i++) {
for (int j = i+1; j < key.length; j++) {
transpose(key, i, j);
}
}
for (int i = 0; i < key.length; i++) {
for (int j = 0; j < key.length/2; j++) {
reverse(key, i, j, key.length-1-j);
}
}
}
public void transpose(int[][] key, int a, int b) {
int temp = key[a][b];
key[a][b] = key[b][a];
key[b][a] = temp;
}
public void reverse(int[][] key, int i, int a, int b) {
int temp = key[i][a];
key[i][a] = key[i][b];
key[i][b] = temp;
}
public boolean move(int[][] key, int[][] lock, int lockCount) {
for (int i = 0; i < key.length + lock.length - 1; i++) {
for (int j = 0; j < key.length + lock.length - 1; j++) {
if (checking(key, i, j, lockCount)) return true;
}
}
return false;
}
public boolean checking(int[][] key, int row, int col, int lockCount) {
int count = lockCount;
for (int i = 0; i < key.length; i++) {
for (int j = 0; j < key[0].length; j++) {
if (isLockSection[i+row][j+col]) {
if (map[i+row][j+col] == key[i][j]) return false;
else if (map[i+row][j+col] == 0 && key[i][j] == 1) {
count--;
}
}
}
}
return count == 0;
}
}
반응형
'알고리즘' 카테고리의 다른 글
[LeetCode] Median of Two Sorted Arrays_배열, 병합 정렬, 힙, 우선순위큐, 이분 탐색 (1) | 2023.12.04 |
---|---|
[알고리즘] 영어 끝말잇기_프로그래머스 레벨2_해시(hash), 나머지 연산(MOD) (0) | 2023.06.26 |
[알고리즘] 이분 탐색(Binary Search) (0) | 2023.05.06 |
[알고리즘] 트리의 지름(백준1967번)_골드4_그래프 탐색, DFS, tree, 그래프와 트리 (0) | 2023.04.17 |
[알고리즘] 빙산(백준2573번)_골드4_그래프 탐색, DFS+BFS (0) | 2023.04.16 |
댓글
반응형
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 타입스크립트
- 자바스크립트
- 스프링
- JPA
- BFS
- JavaScript
- SQLD
- 백준
- 리액트
- 스프링부트
- 해시맵
- java
- 프로그래머스
- 정렬
- 자바
- 형변환
- dfs
- 자바트리
- 이분탐색
- 자바dp
- Comparator
- Nest
- 자바bfs
- Algorithm
- Queue
- SQL
- Spring
- 알고리즘
- CS
- DP
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | |||
5 | 6 | 7 | 8 | 9 | 10 | 11 |
12 | 13 | 14 | 15 | 16 | 17 | 18 |
19 | 20 | 21 | 22 | 23 | 24 | 25 |
26 | 27 | 28 | 29 | 30 | 31 |
글 보관함