[프로그래머스 C++] 무인도 여행

 

[프로그래머스 C++] 무인도 여행

 

https://school.programmers.co.kr/learn/courses/30/lessons/154540

프로그래머스

---

 

 

해결전략

 

너비 우선 탐색, BFS

 

< 핵심전략 >

while(!Q.emtpy()) 가 돌기 전에 방문 체크 if (maps[y][x] != 'X' && ch[y][x] == 0) 를 먼저하여

다음 사이클에서 체크하지 않도록 한다. 

---

 

정답 코드

 

#include <string>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

int dirY[4] = { -1, 0, 1, 0 };
int dirX[4] = { 0, -1, 0, 1 };

queue<pair<int, int>> Q;
int ch[101][101];

vector<int> solution(vector<string> maps) {
    vector<int> answer;

    for (int y = 0; y < maps.size(); y++) {
        for (int x = 0; x < maps[0].size(); x++) 
        {
            int cnt = 0; // 방문지점들의 숫자를 더해주는 변수
            if (maps[y][x] != 'X' && ch[y][x] == 0) 
            {
                Q.push({ y, x }); // 방문지점 Q에 삽입
                ch[y][x] = 1;            // 방문 체크
                cnt += maps[y][x] - '0'; // 방문지점 숫자 더하기

                while (!Q.empty())
                {
                    int y = Q.front().first;
                    int x = Q.front().second;
                    Q.pop();

                    for (int i = 0; i < 4; i++)
                    {
                        int ny = y + dirY[i];
                        int nx = x + dirX[i];

                        if (ny < 0 || maps.size() <= ny || nx < 0 || maps[0].size() <= nx) continue;

                        if (maps[ny][nx] != 'X' && ch[ny][nx] == 0)
                        {
                            Q.push({ ny, nx }); // 새로운 위치 넣기
                            ch[ny][nx] = 1;            // 방문 체크
                            cnt += maps[ny][nx] - '0'; // 방문지점 숫자 더하기
                        }
                    }
                }
                answer.push_back(cnt); // 방문한 그룹의 숫자 합을 answer에 넣음
            }
        }
    }
    

    if (answer.empty()) 
        return vector<int>{-1};

	sort(answer.begin(), answer.end());

    return answer;
}

 

 

---