문제 설명
rows x columns 크기인 행렬이 있습니다. 행렬에는 1부터 rows x columns까지의 숫자가 한 줄씩 순서대로 적혀있습니다. 이 행렬에서 직사각형 모양의 범위를 여러 번 선택해, 테두리 부분에 있는 숫자들을 시계방향으로 회전시키려 합니다. 각 회전은 (x1, y1, x2, y2)인 정수 4개로 표현하며, 그 의미는 다음과 같습니다.
- x1 행 y1 열부터 x2 행 y2 열까지의 영역에 해당하는 직사각형에서 테두리에 있는 숫자들을 한 칸씩 시계방향으로 회전합니다.
다음은 6 x 6 크기 행렬의 예시입니다.
이 행렬에 (2, 2, 5, 4) 회전을 적용하면, 아래 그림과 같이 2행 2열부터 5행 4열까지 영역의 테두리가 시계방향으로 회전합니다. 이때, 중앙의 15와 21이 있는 영역은 회전하지 않는 것을 주의하세요.
행렬의 세로 길이(행 개수) rows, 가로 길이(열 개수) columns, 그리고 회전들의 목록 queries가 주어질 때, 각 회전들을 배열에 적용한 뒤, 그 회전에 의해 위치가 바뀐 숫자들 중 가장 작은 숫자들을 순서대로 배열에 담아 return 하도록 solution 함수를 완성해주세요.
제한사항
- rows는 2 이상 100 이하인 자연수입니다.
- columns는 2 이상 100 이하인 자연수입니다.
- 처음에 행렬에는 가로 방향으로 숫자가 1부터 하나씩 증가하면서 적혀있습니다.
- 즉, 아무 회전도 하지 않았을 때, i 행 j 열에 있는 숫자는 ((i-1) x columns + j)입니다.
- queries의 행의 개수(회전의 개수)는 1 이상 10,000 이하입니다.
- queries의 각 행은 4개의 정수 [x1, y1, x2, y2]입니다.
- x1 행 y1 열부터 x2 행 y2 열까지 영역의 테두리를 시계방향으로 회전한다는 뜻입니다.
- 1 ≤ x1 < x2 ≤ rows, 1 ≤ y1 < y2 ≤ columns입니다.
- 모든 회전은 순서대로 이루어집니다.
- 예를 들어, 두 번째 회전에 대한 답은 첫 번째 회전을 실행한 다음, 그 상태에서 두 번째 회전을 실행했을 때 이동한 숫자 중 최솟값을 구하면 됩니다.
입출력 예시
| rows | columns | queries | result |
| 6 | 6 | [[2,2,5,4],[3,3,6,6],[5,1,6,3]] | [8, 10, 25] |
| 3 | 3 | [[1,1,2,2],[1,2,2,3],[2,1,3,2],[2,2,3,3]] | [1, 1, 5, 3] |
| 100 | 97 | [[1,1,100,97]] | [1] |
입출력 예 설명
입출력 예 #1
- 회전을 수행하는 과정을 그림으로 표현하면 다음과 같습니다.
입출력 예 #2
- 회전을 수행하는 과정을 그림으로 표현하면 다음과 같습니다.
입출력 예 #3
- 이 예시에서는 행렬의 테두리에 위치한 모든 칸들이 움직입니다. 따라서, 행렬의 테두리에 있는 수 중 가장 작은 숫자인 1이 바로 답이 됩니다.
🐊 나의 풀이
import java.util.*;
class Solution {
public int[] solution(int rows, int columns, int[][] queries) {
ArrayList<Integer> min=new ArrayList<>();
int[][] matrix=new int[rows][columns];
int num=1;
for(int r=0; r<rows; r++){
for(int c=0; c<columns; c++){
matrix[r][c]=num;
num++;
}
}
for(int[] query : queries){
int r1=query[0]-1;
int c1=query[1]-1;
int r2=query[2]-1;
int c2=query[3]-1;
int tmp=matrix[r1][c1];
int ans=tmp;
for(int x=r1; x<r2; x++){
matrix[x][c1]=matrix[x+1][c1];
ans=Math.min(ans, matrix[x][c1]);
}
for(int y=c1; y<c2; y++){
matrix[r2][y]=matrix[r2][y+1];
ans=Math.min(ans, matrix[r2][y]);
}
for(int x=r2; x>r1; x--){
matrix[x][c2]=matrix[x-1][c2];
ans=Math.min(ans, matrix[x][c2]);
}
for(int y=c2; y>c1; y--){
matrix[r1][y]=matrix[r1][y-1];
ans=Math.min(ans, matrix[r1][y]);
}
matrix[r1][c1+1]=tmp;
min.add(ans);
}
int[] answer=min.stream().mapToInt(i->i).toArray();
return answer;
}
}다른 사람의 파이썬 풀이를 자바로 구현
# 시간초과 2개
def dfs(rows, columns, query, lst):
copy = [[0] * columns for i in range(rows)]
r1 = query[0] - 1
c1 = query[1] - 1
r2 = query[2] - 1
c2 = query[3] - 1
copy[r1][c1] = lst[r1 + 1][c1] # 상단 맨 왼쪽
copy[r2][c2] = lst[r2 - 1][c2] # 하단 맨 오른쪽
copy[r2][c1]=lst[r2][c1+1] # 하단 맨 왼쪽
result = rows * columns + 1
for x in range(rows):
for y in range(columns):
if copy[x][y] != 0:
result=min(result, copy[x][y])
continue
elif r1 == x and c1 <= y <= c2 : # 상단 -
copy[x][y] = lst[x][y - 1]
elif r2 == x and c1 <= y <= c2 : # 아래 -
copy[x][y]=lst[x][y+1]
elif r1 <= x <= r2 and c1 == y : # 왼쪽 |
copy[x][y] = lst[x+1][y]
elif r1 <= x <= r2 and y == c2 : # 오른쪽 |
copy[x][y] = lst[x-1][y]
else: # 회전 범위에 해당하지 않는 부분
copy[x][y] = lst[x][y]
continue
result = min(result, copy[x][y])
return copy, result
def solution(rows, columns, queries):
answer = []
lst = [[0] * columns for i in range(rows)]
num = 1
# 1부터 rows*columns까지 2차원 리스트 값 초기화
for i in range(rows):
for j in range(columns):
lst[i][j] = num
num += 1
for query in queries:
lst, result = dfs(rows, columns, query, lst)
print(lst)
answer.append(result)
return answer제출하니 2개가 시간초과다.
함수를 호출하면 copy 리스트를 채운 후 lst=copy로 변경 후 query를 또 실행한다.
새 리스트에 값을 채우는 것 보다는 lst 자체에서 회전하게 해야 시간초과가 안 뜰것이다.
# 실패
def dfs(rows, columns, query, lst):
copy = [[0] * columns for i in range(rows)]
r1 = query[0] - 1
c1 = query[1] - 1
r2 = query[2] - 1
c2 = query[3] - 1
copy[r1][c1] = lst[r1 + 1][c1] # 상단 맨 왼쪽
copy[r2][c2] = lst[r2 - 1][c2] # 하단 맨 오른쪽
copy[r2][c1]=lst[r2][c1+1] # 하단 맨 왼쪽
result = rows * columns + 1
for x in range(rows):
for y in range(columns):
if copy[x][y] != 0:
result=min(result, copy[x][y])
continue
elif (r1 == x or x == r2) and c1 <= y <= c2 : # 상단 -와 하단 -
copy[x][y] = lst[x][y - 1]
elif r1 <= x <= r2 and c1 == y : # 왼쪽 |
copy[x][y] = lst[x+1][y]
elif r1 <= x <= r2 and y == c2 : # 오른쪽 |
copy[x][y] = lst[x-1][y]
else: # 회전 범위에 해당하지 않는 부분
copy[x][y] = lst[x][y]
continue
result = min(result, copy[x][y])
return copy, result
def solution(rows, columns, queries):
answer = []
lst = [[0] * columns for i in range(rows)]
num = 1
# 1부터 rows*columns까지 2차원 리스트 값 초기화
for i in range(rows):
for j in range(columns):
lst[i][j] = num
num += 1
for query in queries:
lst, result = dfs(rows, columns, query, lst)
answer.append(result)
return answer문제의 테케는 다 통과했지만 제출하니 1개 통과 ㅠ
쿼리를 실행할 때마다 결과를 출력해봤다.
테케 1의 쿼리 1번째([2,2,5,4])를 실행 후 결과는 아래와 같다.
| 1 | 2 | 3 | 4 | 5 | 6 |
| 7 | 14 | 8 | 9 | 11 | 12 |
| 13 | 20 | 15 | 10 | 17 | 18 |
| 19 | 26 | 21 | 16 | 23 | 24 |
| 25 | 27 | 26 | 22 | 29 | 30 |
| 31 | 32 | 33 | 34 | 35 | 36 |
26이 중복된다. 4행 2열은 28이 맞다.
elif (r1 == x or x == r2) and c1 <= y <= c2 : # 상단 -와 하단 -
copy[x][y] = lst[x][y - 1]이 부분이 잘못되었다.
상단과 하단은 방향이 다르다. 하나로 통합시켰다니 😅
🐝 다른 사람 풀이
# https://roomedia.tistory.com/entry/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%ED%96%89%EB%A0%AC-%ED%85%8C%EB%91%90%EB%A6%AC-%ED%9A%8C%EC%A0%84%ED%95%98%EA%B8%B0-2021-Dev-Matching-%EC%9B%B9-%EB%B0%B1%EC%97%94%EB%93%9C-%EA%B0%9C%EB%B0%9C
def solution(rows, columns, queries) :
matrix=[[row*columns+col+1 for col in range(columns)] for row in range(rows)]
answer=[]
for t, l, b, r in queries :
top, left, bottom, right=t-1, l-1, b-1, r-1
tmp=matrix[top][left]
minimum=tmp
for y in range(top, bottom) : # 왼쪽 |
value=matrix[y+1][left]
matrix[y][left]=value
minimum=min(minimum, value)
for x in range(left, right) : # 아래 -
value=matrix[bottom][x+1]
matrix[bottom][x]=value
minimum=min(minimum, value)
for y in range(bottom, top, -1) : # 오른쪽 |
value=matrix[y-1][right]
matrix[y][right]=value
minimum=min(minimum, value)
for x in range(right, left, -1) : # 위 -
value=matrix[top][x-1]
matrix[top][x]=value
minimum=min(minimum, value)
matrix[top][left+1]=tmp
answer.append(minimum)
return answer왼쪽 상단의 값을 가장 나중에 matrix에 넣는다.
4가지 방향 모두 각각 for문을 돌린다.
방향이 중요하다. 왼쪽, 아래, 오른쪽, 위 순으로 실행한다.
오른쪽 열과 위쪽 행은 인덱스가 큰쪽부터 값을 입력받는다.
for y in range(top, bottom) : # 왼쪽 |
matrix[y][left]=matrix[y+1][left]
minimum=min(minimum, matrix[y][left])나는 value에 matrix 값 안 담고 위처럼 쓰고 싶다.
class Solution {
static int map[][];
public int[] solution(int rows, int columns, int[][] queries) {
int[] answer=new int[queries.length];
map=new int[rows+1][columns+1];
int idx=1;
for(int i=1; i<=rows; i++){
for(int j=1; j<=columns; j++){
map[i][j]=idx++;
}
}
for(int i=0; i<queries.length; i++){
answer[i]=rotate(queries[i][0], queries[i][1], queries[i][2], queries[i][3]);
}
return answer;
}
static int rotate(int x1, int y1, int x2, int y2){
int x=x1;
int y=y1;
int[] dx={0, -1, 0, 1};
int[] dy={1, 0, -1, 0};
int dir=3;
int temp=map[x][y];
int min=temp;
while(true){
if(x == x2 && y == y1){
dir=0;
}
if(x == x2 && y == y2) dir=1; // 위
if(x == x1 && y == y2) dir=2; // 왼쪽
map[x][y]=map[x+dx[dir]][y+dy[dir]];
x+=dx[dir];
y+=dy[dir];
min=Math.min(map[x][y], min);
if(x==x1 && y==y1){ // 다시 원점으로 돌아오면
map[x1][y1+1]=temp;
break;
}
}
return min;
}
}while문 안에서 인덱스로 방향을 조절하고, 다시 원점으로 돌아오면 break한다.
문제 출처 👉 프로그래머스
반응형
'coding test' 카테고리의 다른 글
| [파이썬, Java] 11659. 구간 합 구하기 4 (0) | 2021.10.18 |
|---|---|
| [파이썬, Java] 2661. 좋은수열 (0) | 2021.10.18 |
| [파이썬, Java] 로또의 최고 순위와 최저 순위 (0) | 2021.10.13 |
| [파이썬, Java] 2798. 블랙잭 (0) | 2021.10.11 |
| [파이썬, Java] 11403. 경로 찾기 (0) | 2021.10.08 |