[Algorithm] 1012 유기농 배추

gunggme·2023년 11월 28일

알고리즘

목록 보기
27/42

시작

이 문제는 DFS 깊이 우선 탐색을 이용하여 해결하는 문제다. 그렇다면 우선 예외 사항을 찾아보자.

예외 사항

  1. 범위를 벗어나면 안됨
  2. 그 곳을 방문했으면 안됨
  3. 그 곳에 배추가 안심어 져있으면 안됨

이런 예외사항들이 있는데 이 문제를 풀기에는 충분한 문제다. 한번 간단하게 알고리즘을 짜본 뒤에 코드를 확인 해보자.

알고리즘

  1. 4방향을 우선 확인하기
  2. 만약 그 방향이 밭의 밖이라면 넘기기
  3. 만약 그 방향에 방문을 했다면 넘기기
  4. 만약 그 방향에 배추가 안심어 져있다면 넘기기
  5. 그 방향에 방문했다고 표시

이런식으로 알고리즘을 구성한뒤 코드를 짜면 쉽게 해결되는 문제다. 밑은 코드다

코드

#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<algorithm>
#include<cmath>
#include<tuple>

using namespace std;

int dy[] = {- 1, 0, 1, 0 };
int dx[] = { 0, 1, 0, -1 };
vector<vector<int>> _map;
vector<vector<int>> visited;
int T, N, M, X, Y, count_;

void DFS(int x, int y) {
	// 방문했다고 표시
 	visited[y][x] = 1;

	for (int i = 0; i < 4; i++) {
		int nx = x + dx[i];
		int ny = y + dy[i];
		// 범위를 벗어나는지 확인
		if (nx < 0 || nx >= M || ny < 0 || ny >= N) continue;
		// 방문을 했거나, 그 위치가 배추가 아니면 넘기기
		if (_map[ny][nx] == 0) continue;
		if (visited[ny][nx] == 1) continue;
		DFS(nx, ny);
	}
}

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

	cin >> T;
	for (int i = 0; i < T; i++) {
		int result = 0;
		cin >> M >> N >> count_;
		_map.resize(N);
		visited.resize(N);
		for (int j = 0; j < N; j++) {
			// 모든 칸을 0으로 초기화
			_map[j].assign(M, 0);
			visited[j].assign(M, 0);
		}

		for (int i = 0; i < count_; i++) {
			cin >> X >> Y;
			_map[Y][X] = 1;
		}

		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				// 방문 했는지와, 그 위치에 배추가 있는지 확인
				if (_map[i][j] == 1 && visited[i][j] == 0) {
					result++;
					DFS(j, i);
				}
			}
		}

		cout << result << "\n";
	}

	
} 
profile
안녕하세요!

0개의 댓글