62-9kyo-hwang

[9kyo-hwang]

🔗 문제 링크

무인도 여행

✔️ 소요된 시간

10분

✨ 수도 코드

1. 문제 설명

1 x 1크기의 사각형들로 이루어진 직사각형 격자 형태의 지도 정보가 주어진다. 격자의 각 칸에는 'X' 또는 1에서 9 사이의 자연수가 적혀있다.
지도의 각 칸에서 상, 하, 좌, 우로 연결되는 칸에 적힌 숫자를 모두 합한 값들을 구하고자 한다.

예를 들어, ["X591X","X1X5X","X231X", "1XXX1"]와 같이 문자열로 주어진다면 이는 다음과 같은 지도를 나타내는 것이다.

연결된 칸들의 값을 합치면 다음과 같으며

이는 [1, 1, 27]로 저장할 수 있다.

지도를 나타내는 문자열 배열 maps가 매개변수로 주어질 때, 숫자 칸들을 연결해 얻을 수 있는 숫자들을 배열에 오름차순으로 담아 return 하는 solution 함수를 완성하라. 만약 숫자 칸이 존재하지 않는다면 -1을 배열에 담아 return 하라.

2. 설명

그래프 연결 요소 관련 문제임을 알 수 있다. 연결요소 관련된 문제는 여러 유형이 있으며, 여기서는 각 연결요소의 '크기'를 구하는 문제로 해석할 수 있다.
각 칸(즉, 그래프 노드)의 크기가 1 ~ 9 사이이고, 상/하/좌/우로 노드가 연결된 그래프로 해석할 수 있다.

문제 해석이 완료됐다면, 코드는 쉽게 구현할 수 있다.

vector<int> DaysofStay;
for(int i = 0; i < Maps.size(); ++i)
{
    for(int j = 0; j < Maps[0].size(); ++j)
    {
        if(Maps[i][j] != 'X')
        {
            DaysofStay.emplace_back(Merge(Maps, i, j));
        }
    }
}

무인도가 아닌 칸(노드)이라면, Merge 함수를 호출해 해당 함수가 반환하는 연결 요소의 크기를 DaysofStay 리스트에 저장한다.

int Merge(vector<string>& Maps, int x, int y)
{
    if(x < 0 || x >= Maps.size() || y < 0 || y >= Maps[0].size() || Maps[x][y] == 'X') 
    {
        return 0;
    }
    
    int Day = Maps[x][y] - '0';
    Maps[x][y] = 'X';
    
    return Day 
        + Merge(Maps, x - 1, y) 
        + Merge(Maps, x, y + 1) 
        + Merge(Maps, x + 1, y) 
        + Merge(Maps, x, y - 1);
}

연결요소의 크기를 반환하는 Merge 함수는 DFS를 기반으로 작성했다.
현재 (x, y) 좌표가 그리드 범위를 벗어났거나 무인도('X') 칸이라면 0을 반환하고, 아니라면 현재 칸의 숫자(Maps[x][y], char이므로 int로 변환 필요)와 인접한 4방향 칸에 대한 재귀 호출의 결과를 모두 더해 반환한다.
이 때, 현재 칸엔 더 이상 방문하지 않도록 마킹(무인도, 즉 'X' 값으로 변경)해준다.

if(DaysofStay.empty()) 
{
    return {-1};
}

sort(DaysofStay.begin(), DaysofStay.end());
return DaysofStay;

지도의 모든 칸에 대해 검사를 완료했는데도 DaysofStay 리스트가 비어있다면, 숫자 칸이 존재하지 않는 것이므로 [-1]을 반환한다.
아니라면 오름차순 정렬 후 반환한다.

전체 코드

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

using namespace std;

int Merge(vector<string>& Maps, int x, int y)
{
    if(x < 0 || x >= Maps.size() || y < 0 || y >= Maps[0].size() || Maps[x][y] == 'X') 
    {
        return 0;
    }
    
    int Day = Maps[x][y] - '0';
    Maps[x][y] = 'X';
    
    return Day 
        + Merge(Maps, x - 1, y) 
        + Merge(Maps, x, y + 1) 
        + Merge(Maps, x + 1, y) 
        + Merge(Maps, x, y - 1);
}

vector<int> solution(vector<string> Maps) 
{
    vector<int> DaysofStay;
    for(int i = 0; i < Maps.size(); ++i)
    {
        for(int j = 0; j < Maps[0].size(); ++j)
        {
            if(Maps[i][j] != 'X')
            {
                DaysofStay.emplace_back(Merge(Maps, i, j));
            }
        }
    }
    
    if(DaysofStay.empty()) 
    {
        return {-1};
    }
    
    sort(DaysofStay.begin(), DaysofStay.end());
    return DaysofStay;
}

📚 새롭게 알게된 내용

음 이번에도 너무 빨리 풀어버린...

[xxubin04]

10분만에 푸시다니...
무인도칸이거나 범위를 벗어나면 0을 반환해서 쉽게 더할 수 있군요!
수고 많으셨습니다..!

[mjj111]

무지성 bfs 바로 때렸습니다.

import collections

def can_go(x, y):
    if not (0 <= x < N and 0 <= y < M):
        return False
    if visited[x][y]:
        return False
    if maps[x][y] == 'X':
        return False
    return True

def get_food_amount(start_x, start_y):
    global dx, dy, maps, visited
    
    food = 0
    queue = collections.deque([(start_x, start_y)]) 
    visited[start_x][start_y] = True 
    
    while queue:
        x, y = queue.popleft()
        food += int(maps[x][y])

        for i in range(4):
            next_x = x + dx[i]
            next_y = y + dy[i]
            if can_go(next_x, next_y):
                visited[next_x][next_y] = True
                queue.append((next_x, next_y))
    
    return food

N, M = 0, 0
dx = [0, 0, 1, -1]
dy = [1, -1, 0, 0]
maps, visited = [], []

def solution(mapss):
    global N, M, visited, maps
    maps = mapss
    N = len(maps)
    M = len(maps[0])
    visited = [[False] * M for _ in range(N)]
    
    answer = []
    for i in range(N):
        for j in range(M):
            if maps[i][j] != 'X' and not visited[i][j]:
                answer.append(get_food_amount(i, j))
    
    if not answer:
        return [-1]
    
    return sorted(answer)

[gjsk132]

오... mapss 까리한데요

[gjsk132]

파이썬은 문자열 요소 변경이 안되서, 따로 체크할 수 있도록 처리해주어야하는데...

visited(방문체크)를 만들거나 maps를 산산히 분해하는 방법이 있습니다...

저는 maps를 하나씩 분리시켜줬습니다!

코드

def solution(maps):
    
    answer = []    
    offset = [(-1, 0), (0, 1), (1, 0), (0, -1)]

    # 여기서 maps을 분리해주고, X는 0으로 치환해줬습니다.
    memo = [[int(i) if not i == 'X' else 0 for i in row] for row in maps]

    h, w = len(maps), len(maps[0])

    def bfs(x, y):
        total = memo[x][y]
        memo[x][y] = 0

        dq = [(x, y)]

        while dq:
            x, y = dq.pop()

            for dx, dy in offset:
                px = x + dx
                py = y + dy

                if not 0 <= px < h or not 0 <= py < w:
                    continue

                if (value := memo[px][py]) == 0:
                    continue

                total += value
                memo[px][py] = 0
                dq.append((px, py))

        return total

    for x in range(h):
        for y in range(w):
            if memo[x][y] == 0:
                continue
            
            answer.append(bfs(x, y))

    if not answer:
        answer.append(-1)

    return sorted(answer)