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)입니다.
예를 들어, 두 번째 회전에 대한 답은 첫 번째 회전을 실행한 다음, 그 상태에서 두 번째 회전을 실행했을 때 이동한 숫자 중 최솟값을 구하면 됩니다.
풀이
입출력 예시
풀이
행렬부터 만들고 고민하기 위해 먼저 행렬을 생성해주는 코드부터 짰다. 행렬은 2차원 벡터로 선언해 줘도 되지만, 나는 그냥 1차원으로 선언해줬다. 행렬의 크기는 rows x columns 이므로, rows x columns 크기의 벡터를 만들었다.
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> solution(int rows, int columns, vector<vector<int>> queries) {
vector<int> answer;
vector<int> matrix;
int m = min(rows, columns); // rows와 columns중 더 작은 값
for (int i = 1; i <= rows * columns; i++) {
matrix.push_back(i);
}
return answer;
}
1차원 벡터이므로 2차원과는 조금 다른 방식으로 접근해야한다. row가 r이고 columns가 c인 행렬이 있다고 하자. 그러면 2차원 벡터는 matrix[r][c] 라면, 1차원 벡터는 matrix[r*c]가 된다. (i, j)에 접근하기 위해서는 2차원 벡터는 matrix[i][j]로 접근한다면, 1차원 벡터는 matrix[i * c + j] 가 된다.
이제 회전시켜보자. 회전은 오른쪽으로 (y2 - y1)번씩 이동하고, 아래로 (x2 - x1)번씩, 왼쪽으로 (y2 - y1)번씩, 위로 (x2 - x1)번씩 이동한다.
...
...
for (const auto& q : queries) {
// 2차원 행렬을 [x][y]라고 할 때
int x1y1 = (q[0] - 1) * m + (q[1] - 1);
int x1y2 = (q[0] - 1) * m + (q[3] - 1);
int x2y1 = (q[2] - 1) * m + (q[1] - 1);
int x2y2 = (q[2] - 1) * m + (q[3] - 1);
int min_num = 99999999;
int temp = matrix[x1y1];
min_num = min(min_num, temp);
// 오른쪽으로 1번씩 이동
for (int i = x1y1 + 1; i <= x1y2; i++) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
// 아래로 1번씩 이동
for (int i = x1y2 + m; i <= x2y2; i += m) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
// 왼쪽으로 1번씩 이동
for (int i = x2y2 - 1; i >= x2y1; i--) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
// 위로 1번씩 이동
for (int i = x2y1 - m; i >= x1y1; i -= m) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
answer.push_back(min_num);
}
...
...
이동한 숫자 중에서 최소를 구하는 것이므로 이동을 할 때마다 최솟값을 계속 갱신해 줬다.
전체 코드
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> solution(int rows, int columns, vector<vector<int>> queries) {
vector<int> answer;
vector<int> matrix;
for (int i = 1; i <= rows * columns; i++) {
matrix.push_back(i);
}
for (const auto& q : queries) {
// 2차원 행렬을 [x][y]라고 할 때
int x1y1 = (q[0] - 1) * columns + (q[1] - 1);
int x1y2 = (q[0] - 1) * columns + (q[3] - 1);
int x2y1 = (q[2] - 1) * columns + (q[1] - 1);
int x2y2 = (q[2] - 1) * columns + (q[3] - 1);
int min_num = 99999999;
int temp = matrix[x1y1];
min_num = min(min_num, temp);
// 오른쪽으로 1번씩 이동
for (int i = x1y1 + 1; i <= x1y2; i++) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
// 아래로 1번씩 이동
for (int i = x1y2 + columns; i <= x2y2; i += columns) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
// 왼쪽으로 1번씩 이동
for (int i = x2y2 - 1; i >= x2y1; i--) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
// 위로 1번씩 이동
for (int i = x2y1 - columns; i >= x1y1; i -= columns) {
swap(matrix[i], temp);
min_num = min(min_num, temp);
}
answer.push_back(min_num);
}
return answer;
}
