[BOJ] 3184 양

문제보기

맥북 충전기를 두고와서 

윈도우인 집 컴퓨터에서 웹 IDE를 사용해서 풀다보니 디버깅이 어려워서

엉뚱하게 입력부분에서 헤멨던 문제.

BFS를 사용해서 구간별로 늑대와 양의 수를 구한 후 

양이 많으면 양의 수를 리턴,

늑대가 많으면 늑대 수 * -1을 해준다.

메인에서 각각 양의수와 늑대 수 (*-1)로 답을 도출한다.

---

#include <queue>

using namespace std;

char map[251][251]={};

int check[251][251]={};

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

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

int N,M;

int BFS(pair<int, int> n){

queue<pair<int, int> > q;

q.push(n);

int wolf=0;

int sheep=0;

while(!q.empty()){

int y = q.front().first;

int x = q.front().second;

q.pop();

char cur = map[y][x];

if(cur == 'v'){

wolf++;

}else if(cur == 'o'){

sheep++;

}

check[y][x] = 1;

for(int i = 0;i<4;i++){

int ny = y+dy[i];

int nx = x+dx[i];

if(nx >=0 && nx <M && ny>=0 && ny <N){

if(check[ny][nx] == 0 && map[ny][nx]!='#'){

q.push(make_pair(ny,nx));

check[ny][nx] = 1;

}

}

}

}

if(sheep > wolf){

return sheep;

}else{

return wolf * -1;

}

}

int main(){

scanf("%d%d",&N,&M);

for(int i = 0;i<N;i++){

char temp[251];

scanf("%s",temp);

for(int j = 0;j<M;j++){

    map[i][j] = temp[j];

}

}

int ansWolf = 0;

int ansSheep = 0;

for(int i = 0;i<N;i++){

for(int j = 0;j<M;j++){

if(check[i][j] == 0 && map[i][j] != '#'){

int tp = BFS(make_pair(i,j));

if(tp > 0){

ansSheep+=tp;

}else{

ansWolf+=tp*-1;

}

}

}

}

printf("%d %d\n",ansSheep,ansWolf);

}