코딩테스트 문제풀이/백준

[백준][BFS] 토마토

itaeiou 2020. 3. 1. 20:44
반응형

백준 알고리즘

BFS

C++

 

https://www.acmicpc.net/problem/7576

 

7576번: 토마토

첫 줄에는 상자의 크기를 나타내는 두 정수 M,N이 주어진다. M은 상자의 가로 칸의 수, N은 상자의 세로 칸의 수를 나타낸다. 단, 2 ≤ M,N ≤ 1,000 이다. 둘째 줄부터는 하나의 상자에 저장된 토마토들의 정보가 주어진다. 즉, 둘째 줄부터 N개의 줄에는 상자에 담긴 토마토의 정보가 주어진다. 하나의 줄에는 상자 가로줄에 들어있는 토마토의 상태가 M개의 정수로 주어진다. 정수 1은 익은 토마토, 정수 0은 익지 않은 토마토, 정수 -1은 토마

www.acmicpc.net

 

#include <iostream>
#include <queue>
using namespace std;

int M, N;
int map[1000][1000];
int visited[1000][1000] = { 0, };

int dx[4] = { -1, 0, 1, 0 };
int dy[4] = { 0, -1, 0, 1 };
queue<pair<int, int>> q;
int cnt=0;

void bfs();

int main()
{
	cin >> M >> N;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			cin >> map[i][j];
		}
	}

	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (map[i][j] == 1) {
				q.push(make_pair(i, j));
				//visited[i][j] = 1;
			}			
		}
	}

	bfs();

	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (map[i][j] == 0) {
				cout << -1;
				return 0;
			}
		}
	}
	cout << cnt;
	return 0;
}

void bfs() {
	int x, y, nx, ny;
	while (!q.empty())
	{
		x = q.front().first;
		y = q.front().second;
		q.pop();
		for (int k = 0; k < 4; k++) {
			nx = x + dx[k];
			ny = y + dy[k];
			if (nx >= 0 && nx < N && ny >= 0 && ny < M) {
				if (map[nx][ny] == 0 && visited[nx][ny] == 0) {
					map[nx][ny] = 1;
					visited[nx][ny] = visited[x][y] + 1;
					cnt = visited[nx][ny];
					q.push(make_pair(nx, ny));
				}
			}
		}		
	}
}
반응형

'코딩테스트 문제풀이 > 백준' 카테고리의 다른 글

[백준] 별 찍기 - 10  (0) 2021.04.06
[백준][DFS] 적록색약  (0) 2020.03.01
[백준][DFS] 경로찾기  (0) 2020.03.01