[백준](Java)1303 - 전쟁 - 전투

문제 링크

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

문제 풀이

BFS 뼈대 방식을 활용하여 접근했고
bfs에서 지나친 칸의 갯수를 카운트 하고, 그값을 return 하고 제곱한뒤에 B,W의 값을 담아두는 변수에 + 해준다.

코드

import java.util.*;

public class Main {

    static int n;
    static int m;
    static int [] diry ={-1,0,1,0}; //상 우 하 좌
    static int [] dirx ={0,1,0,-1};

    static String [][] map;
    static boolean [][] chk;

    static int white_cnt =0;
    static int black_cnt =0;
    static class Node{
        int y;
        int x;

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


    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);

        n = sc.nextInt();
        m = sc.nextInt();

        map = new String[m][n];
        chk = new boolean[m][n];


        for(int i = 0; i<m; i++){
            String line = sc.next();
            for(int j =0; j<n; j++){
                map[i][j] = line.substring(j,j+1);
            }
        }

        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if(!chk[i][j]){
                    if(map[i][j].equals("B")){
                        int num = bfs(i,j,"B");
                        black_cnt += num*num;
                    }else{
                        int num = bfs(i,j,"W");
                        white_cnt += num*num;
                    }

                }
            }
        }
        System.out.println(white_cnt+" "+black_cnt);
        //System.out.println(Arrays.deepToString(map));
    }

    public static int bfs(int y, int x, String color){
        //System.out.print(y+" "+x+" "+color);
        Queue<Node>queue  = new LinkedList<>();
        queue.add(new Node(y,x));
        int cnt =1;
        chk[y][x] = true;
        while(!queue.isEmpty()){
            Node now = queue.poll();
            for(int i=0; i<4; i++){
                int now_y = now.y+diry[i];
                int now_x = now.x+dirx[i];
                if(now_y>=0 && now_y<m && now_x>=0 && now_x<n){
                    if(!chk[now_y][now_x] && color.equals(map[now_y][now_x])){
                        queue.add(new Node(now_y,now_x));
                        chk[now_y][now_x] = true;
                        cnt++;
                    }
                }

            }
        }
        //System.out.println(" "+cnt);
        return cnt;
    }

}