백준 1743 음식물 피하기 (C++)

안유태·2023년 8월 22일
0

알고리즘

목록 보기
133/239
post-custom-banner

1743번: 음식물 피하기

bfs를 이용한 문제이다. 풀이는 간단하다. 입력받은 음식물 위치들을 반복문을 돌면서 bfs를 돌려준다. bfs를 통해 연속으로 이어진 음식물의 개수를 카운트하여 이를 result와 비교하여 큰 값을 저장해주었다. 그 후 result를 출력해주었다. 간단하게 풀 수 있었던 문제였다.



#include <iostream>
#include <vector>
#include <queue>
#include <cstring>

using namespace std;

int N, M, K;
vector<pair<int, int>> v;
bool trash[101][101];
bool check[101][101];
int dy[4] = { -1,1,0,0 };
int dx[4] = { 0,0,-1,1 };
int result = 0;

void bfs(int a, int b) {
    queue<pair<int, int>> q;
    int count = 1;

    q.push({ a,b });
    check[a][b] = true;

    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 + dy[i];
            int nx = x + dx[i];

            if (ny < 0 || nx < 0 || ny > N || nx > M) continue;
            if (check[ny][nx]) continue;
            if (!trash[ny][nx]) continue;

            q.push({ ny,nx });
            check[ny][nx] = true;
            count++;
        }
    }

    result = max(result, count);
}

void solution() {
    for (int i = 0; i < K; i++) {
        memset(check, false, sizeof(check));
        bfs(v[i].first, v[i].second);
    }

    cout << result;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);

    cin >> N >> M >> K;

    int a, b;
    for (int i = 0; i < K; i++) {
        cin >> a >> b;
        v.push_back({ a,b });
        trash[a][b] = true;
    }

    solution();

    return 0;
}
profile
공부하는 개발자
post-custom-banner

0개의 댓글