[알고리즘]_BFS/DFS_단지번호 붙이기(백준 No.2667)

문제

<그림 1>과 같이 정사각형 모양의 지도가 있다. 1은 집이 있는 곳을, 0은 집이 없는 곳을 나타낸다. 철수는 이 지도를 가지고 연결된 집의 모임인 단지를 정의하고, 단지에 번호를 붙이려 한다. 여기서 연결되었다는 것은 어떤 집이 좌우, 혹은 아래위로 다른 집이 있는 경우를 말한다. 대각선상에 집이 있는 경우는 연결된 것이 아니다. <그림 2>는 <그림 1>을 단지별로 번호를 붙인 것이다. 지도를 입력하여 단지수를 출력하고, 각 단지에 속하는 집의 수를 오름차순으로 정렬하여 출력하는 프로그램을 작성하시오.

---

입력

첫 번째 줄에는 지도의 크기 N(정사각형이므로 가로와 세로의 크기는 같으며 5≤N≤25)이 입력되고, 그 다음 N줄에는 각각 N개의 자료(0혹은 1)가 입력된다.

---

출력

첫 번째 줄에는 총 단지수를 출력하시오. 그리고 각 단지내 집의 수를 오름차순으로 정렬하여 한 줄에 하나씩 출력하시오.

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
import java.util.Queue;

public class No_2667 {
	static int[] ax = {1, 0, -1, 0};//우하좌상
	static int[] ay = {0, 1, 0, -1};
	static boolean visited[][];
	static int map[][];

	public static void main(String[] args) throws IOException {
		Scanner sc=new Scanner(System.in);
		int N = sc.nextInt();
		map = new int[N][N];
		visited = new boolean[N][N];
		ArrayList<Integer>apart=new ArrayList<>();

		for (int i = 0; i < N; i++) {
			String input=sc.next();
			for (int j = 0; j < N; j++) {
				map[i][j] = input.charAt(j)-'0';
			}
		}
		int count = 0;
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < N; j++) {
				if (map[i][j] == 1 && !visited[i][j]) {
					apart.add(BFS(i, j));
					count++;
				}
			}
		}
		Collections.sort(apart);
		System.out.println(count);
		for (int i:apart) {
			System.out.println(i);
		}
	}

	private static int BFS(int i, int j) {
		Queue<Node> queue = new LinkedList<>();
		queue.add(new Node(i, j));

		visited[i][j] = true;
		int number=1; //처음 들어온 숫자도 있으니까 1부터 시작
		while (!queue.isEmpty()) {
			Node now = queue.poll();

			for (int k = 0; k < 4; k++) {
				int nextX = now.x + ax[k];
				int nextY = now.y + ay[k];
				if (nextX >= 0 && nextY >= 0 && nextX < visited.length && nextY < visited.length){
					if(!visited[nextX][nextY]){
						if(map[nextX][nextY]==1){
							queue.add(new Node(nextX,nextY));
							visited[nextX][nextY]=true;
							number++;
						}
					}
				}
			}
		}
		return number;
	}

	static class Node {
		int x;
		int y;

		public Node(int x, int y) {
			this.x = x;
			this.y = y;
		}

	}
}

풀이

• 유기농 배추와 푸는 방법이 유사하다
• BFS 메서드에서 number 변수는 아파트 단지 수를 세는 용도이다
• 처음에 매개변수로 들어온 좌표도 아파트가 있는 곳이니 1부터 시작한다!

마무리

처음으로 혼자 그래프 문제 생각해보고 유기농 배추 참고해서 풀어봤다
뭔말인지 알겠어서 좀 재밌게 풀었다 사실 추가한거 별로 없음 그래도 이해한대로 풀어봤당
코드 테스트 돌려보기가 너무 싫어서 ㅋ ㅋㅋ ㅋ .. 코드 맞게 짠거같아소 고냥 돌렸는데 계속 런타임에러나서 뭔가 봤더니 입력에서 공백을 안주고 그냥 입력값을 준거였다 휴 이거땜에 30분 넘게 엄한거 건드림
글고 해결해서 다시 제출했는데 오류났다 number변수를 0부터 시작해서 틀렸었따 껄껄..