백준 1388: 바닥 장식 [Java] - 포포

2022. 5. 16. 22:36알고리즘/BFS 알고리즘

문제

형택이는 건축가이다. 지금 막 형택이는 형택이의 남자 친구 기훈이의 집을 막 완성시켰다. 형택이는 기훈이 방의 바닥 장식을 디자인했고, 이제 몇 개의 나무 판자가 필요한지 궁금해졌다. 나무 판자는 크기 1의 너비를 가졌고, 양수의 길이를 가지고 있다. 기훈이 방은 직사각형 모양이고, 방 안에는 벽과 평행한 모양의 정사각형으로 나누어져 있다.

이제 ‘-’와 ‘|’로 이루어진 바닥 장식 모양이 주어진다. 만약 두 개의 ‘-’가 인접해 있고, 같은 행에 있다면, 두 개는 같은 나무 판자이고, 두 개의 ‘|’가 인접해 있고, 같은 열에 있다면, 두 개는 같은 나무 판자이다.

기훈이의 방 바닥을 장식하는데 필요한 나무 판자의 개수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 방 바닥의 세로 크기N과 가로 크기 M이 주어진다. 둘째 줄부터 N개의 줄에 M개의 문자가 주어진다. 이것은 바닥 장식 모양이고, '-‘와 ’|‘로만 이루어져 있다. N과 M은 50 이하인 자연수이다.

출력

첫째 줄에 문제의 정답을 출력한다.

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

 


내 제출

package com.company.ch9BFS;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class No1388 {
    static int n,m;
    static char[][] arr;
    static boolean[][] visit;
    static int[] dx = {0, 0, 1, -1};
    static int[] dy = {1, -1, 0, 0};
    static Queue<Point> qu;
    static int count;
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken(" "));
        m = Integer.parseInt(st.nextToken(" "));

        arr = new char[n][m];
        visit = new boolean[n][m];
        qu = new LinkedList<>();
        count = 0;

        for (int i = 0; i < n; i++) {
            String s = br.readLine();
            for (int j = 0; j < m; j++) {
                arr[i][j] = s.charAt(j);
            }
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if(!visit[i][j]){
                    bfs(new Point(i,j));
                    count++;
                }
            }
        }
        System.out.println(count);
    }
    private static void bfs(Point point) {
        qu.add(point);
        visit[point.x][point.y] = true;

        while(!qu.isEmpty()){
            Point p = qu.poll();

            for (int i = 0; i < 4; i++) {
                int nx = p.x+dx[i];
                int ny = p.y+dy[i];

                if(nx<0 || ny<0 || nx>=n || ny>=m) continue;

                if(arr[p.x][p.y] == '-'){
                    if(arr[p.x][ny] == arr[p.x][p.y] && !visit[p.x][ny]){
                        qu.offer(new Point(p.x, ny));
                        visit[p.x][ny] = true;
                    }
                }else{
                    if(arr[nx][p.y] == arr[p.x][p.y] && !visit[nx][p.y]){
                        qu.offer(new Point(nx, p.y));
                        visit[nx][p.y] = true;
                    }
                }
            }
        }
    }

    static class Point {
        int x;
        int y;

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

타일이 '-' 인 경우, bfs를 가로 방향으로,
'l' 인 경우는 세로 방향으로만 실시한 뒤, 각각의 개수 합을 count로 출력하였다.

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