알고리즘/PS - 백준

[백준 3184 - C언어] 양: BFS

excited-hyun 2021. 1. 27. 17:42
반응형

www.acmicpc.net/problem/3184

 

3184번: 양

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다. 다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

www.acmicpc.net

문제

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.

마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.

한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.

다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.

맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.

아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

입력

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.

다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

출력

하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.

 

 

풀이

매우 기본적인 BFS 문제다.

전체 map을 나타내는 2차원 배열과 방문여부를 확인하기 위한 2차원 배열 2가지를 이용한다.

main에서 2중 for문 안에서 방문 되지 않은 칸에 대해서 만 bfs를 수행하면된다.

왜냐하면 한 칸에 대해 bfs를 거치고 나면 그 칸이 속한  울타리 안 영역은 모두 방문한 상태가 되기 때문!

 

이 경우에서도 bfs함수에선 이전 문제들과 마찬가지로 map의 범위를 벗어나는지, 이미 방문한 곳인지, 울타리인지 중요하다!

상하좌우로 움직이면서 차례로 큐에 넣고 빼면서 양인지 늑대인지 확인하고

큐가 텅 비게 되면 양과 늑대의 수를 비교하기만 하면 끝! 거리 안구해도 된다는게 얼마나 좋은지...

 

이전 bfs 문제들에 비해 쉬워서 살것같다

#include <stdio.h>
#include <stdlib.h>

char map[250][250];
int visited[250][250];
int wolf=0;
int sheep=0;

int total_w=0;
int total_s=0;

int x[4] = {1, -1, 0, 0};
int y[4] = {0, 0, 1, -1};

struct Node {
    int x;
    int y;
    struct Node* next;
};

void bfs(struct Node * head, int r, int c);

int main(void){
    
    int r, c;
    struct Node * head;
    struct Node * new;
    scanf("%d %d", &r, &c);
    
    for (int i=0; i< r; i++){
        for(int j=0; j<c; j++){
            scanf("%c", &map[i][j]);
        }
        getchar();
    }
    
    head = (struct Node*) malloc(sizeof(struct Node));
    for(int i=0; i<r; i++){
        for(int j=0; j<c; j++){
            if(visited[i][j] == 1)
                continue;
            new = (struct Node*) malloc(sizeof(struct Node));
            new->x = i;
            new->y = j;
            new->next = NULL;
            head->next = new;
            bfs(head, r, c);
            wolf = 0;
            sheep = 0;
        }
    }
    printf("%d %d\n", total_s, total_w);
}

void bfs(struct Node * head, int r, int c){
    struct Node* temp, *new, *use;
    int X, Y;
    temp = head->next;
    use = temp;
    while(head->next != NULL){
        temp = head->next;
        for(int i=0; i<4; i++){
            X = temp->x + x[i];
            Y = temp->y + y[i];
            
            if( X< 0 || X>=r || Y<0 || Y>=c)
                continue;
            if(visited[X][Y] == 1 || map[X][Y] == '#')
                continue;
            
            visited[X][Y] = 1;
            if(map[X][Y] == 'v'){   //늑대
                wolf++;
            }
            if(map[X][Y] == 'o'){   //양
                sheep++;
            }
            new = (struct Node*) malloc(sizeof(struct Node));
            new->x = X;
            new->y = Y;
            new->next = NULL;
            use->next = new;
            use = use->next;
        }
        head->next = temp->next;
        free(temp);
    }
    if(sheep > wolf)
        total_s += sheep;
    else
        total_w += wolf;
}
728x90
반응형