코딩 테스트 일기 - 4일 차

무인도 여행

문제 링크 : 무인도 여행

문제 설명

메리는 여름을 맞아 무인도로 여행을 가기 위해 지도를 보고 있습니다. 지도에는 바다와 무인도들에 대한 정보가 표시돼 있습니다. 지도는 1 x 1크기의 사각형들로 이루어진 직사각형 격자 형태이며, 격자의 각 칸에는 'X' 또는 1에서 9 사이의 자연수가 적혀있습니다. 지도의 'X'는 바다를 나타내며, 숫자는 무인도를 나타냅니다. 이때, 상, 하, 좌, 우로 연결되는 땅들은 하나의 무인도를 이룹니다. 지도의 각 칸에 적힌 숫자는 식량을 나타내는데, 상, 하, 좌, 우로 연결되는 칸에 적힌 숫자를 모두 합한 값은 해당 무인도에서 최대 며칠동안 머물 수 있는지를 나타냅니다. 어떤 섬으로 놀러 갈지 못 정한 메리는 우선 각 섬에서 최대 며칠씩 머물 수 있는지 알아본 후 놀러갈 섬을 결정하려 합니다.

지도를 나타내는 문자열 배열 maps가 매개변수로 주어질 때, 각 섬에서 최대 며칠씩 머무를 수 있는지 배열에 오름차순으로 담아 return 하는 solution 함수를 완성해주세요. 만약 지낼 수 있는 무인도가 없다면 -1을 배열에 담아 return 해주세요.

제한 사항

• 3 ≤ maps의 길이 ≤ 100
  -> 3 ≤ maps[i]의 길이 ≤ 100
  -> maps[i]는 'X' 또는 1 과 9 사이의 자연수로 이루어진 문자열입니다.
  -> 지도는 직사각형 형태입니다.

입출력 예

maps	result
["X591X","X1X5X","X231X", "1XXX1"]	[1, 1, 27]
["XXX","XXX","XXX"]	[-1]

풀이 방법

• maps를 순회하면서 숫자를 발견하면, 그 숫자를 기점으로 dfs를 통해 섬 전체의 크기를 구하고자 한다. 그리고 string으로 관리되기 때문에 섬 크기에 더한 후에 X로 치환해서 방문여부를 관리하고자 한다.
  -> dfs 알고리즘으로 인해 O(N*M)의 시간복잡도가 발생할 것으로 보이며, 제한사항에 maps의 길이가 100이기 때문에 알맞은 알고리즘으로 보인다.
  
  

구현

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

// dfs로 갈 수 있는지 검사
int dfs(int y, int x, vector<string>& maps){
    int sum = maps[y][x] - '0';
    maps[y][x] = 'X';
    
    // 상
    if(y + 1 < maps.size() && maps[y+1][x] != 'X'){
        sum += dfs(y + 1, x, maps);
    }
    // 하
    if(y - 1 >= 0 && maps[y-1][x] != 'X'){
        sum += dfs(y - 1, x, maps);
    }
    // 좌
    if(x - 1 >= 0 && maps[y][x-1] != 'X'){
        sum += dfs(y, x - 1, maps);
    }
    // 우
    if(x + 1 < maps[0].size() && maps[y][x+1] != 'X'){
        sum += dfs(y, x + 1, maps);
    }
    
    return sum;
}


vector<int> solution(vector<string> maps) {
    vector<int> answer;
    
    for(int i = 0; i < maps.size(); i++){
        for(int j = 0; j < maps[0].size(); j++){
            if(maps[i][j] != 'X'){
                answer.push_back(dfs(i, j, maps));
            }
        }
    }
    
    // 지낼 수 있는 무인도가 없는 경우
    if(answer.empty()){
        answer.push_back(-1);
    }else{
        sort(answer.begin(), answer.end());
    }
    
    return answer;
}

---

행렬 테두리 회전하기

문제 링크 : 행렬 테두리 회전하기

문제 설명

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, 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]

풀이 방법

• 먼저 rows, columns의 개수에 따른 테이블을 생성한다.
• queries에 들어있는 값에 따라 회전을 시켜야 하는데, 범위의 모든 값이 회전하는 것이 아닌 테두리의 값만 회전을 하게 된다. 일단 한 칸씩 밀어야 되기 때문에 임의의 값을 빼놓고, 순서대로 위로 밀고, 왼쪽으로 밀고, 아래로 밀고, 오른쪽으로 미는 과정을 거치면서 최소값 검사까지 하면 될 것으로 보인다.
  -> rows 값을 N, columns 값을 M, queries 행의 개수를 Q라고 했을 때, O((N+M) * Q)로 약 10,000 * 100 번 실행될 것으로 보여서 적당한 알고리즘으로 보인다.
  
  

구현

#include <string>
#include <vector>

using namespace std;

// 시계방향으로 회전시키는 함수
int rotate(vector<vector<int>>& table, int y1, int x1, int y2, int x2){
    int small = 10000;
    int k = table[y1][x1];  // 임시 저장
     
    // 위로 밀기
    for(int i = y1 + 1; i <= y2; i++){
        small = min(small, table[i][x1]);
        table[i - 1][x1] = table[i][x1];
    }
    // 왼쪽으로 밀기
    for(int i = x1 + 1; i <= x2; i++){
        small = min(small, table[y2][i]);
        table[y2][i - 1] = table[y2][i];
    }
    // 아래로 밀기
    for(int i = y2 - 1; i >= y1; i--){
        small = min(small, table[i][x2]);
        table[i + 1][x2] = table[i][x2];
    }
    // 오른쪽으로 밀기
    for(int i = x2 - 1; i >= x1; i--){
        small = min(small, table[y1][i]);
        table[y1][i + 1] = table[y1][i];
    }
    
    small = min(small, k);
    table[y1][x1 + 1] = k;
    
    return small;
}

vector<int> solution(int rows, int columns, vector<vector<int>> queries) {
    vector<int> answer;
    
    // 초기 테이블 세팅
    vector<vector<int>> table(rows + 1, vector<int>(columns + 1));
    
    for(int i = 1; i <= rows; i++){
        for(int j = 1; j <= columns; j++){
            table[i][j] = (i - 1) * columns + j;
        }
    }
    
    // 회전 목록에 따라 회전시키기
    for(vector<int> v : queries){
        answer.push_back(rotate(table, v[0], v[1], v[2], v[3]));
    }
    
    return answer;
}