[Java] 백준 1303번 : 전쟁 - 전투

1. 문제

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

1303번: 전쟁 - 전투

 

 

2. 풀이 과정

dfs로 같은 팀끼리 인접한 수를 구하는데, 방문체크는 W, B => . 으로 변경하여 확인하였다.

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
    static int N, M, result, dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
    static char map[][];

    public static void dfs(int x, int y, char team){
        map[x][y] = '.';
        result++;
        for(int i=0; i<4; i++){
            int nx = x + dx[i];
            int ny = y + dy[i];
            if(nx<0 || nx>=N || ny<0 || ny>=M || map[nx][ny] != team){
                continue;
            }
            dfs(nx, ny, team);
        }
    }

    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String inputs[] = br.readLine().split(" ");
        M = Integer.parseInt(inputs[0]);
        N = Integer.parseInt(inputs[1]);
        map = new char[N][M];

        for (int i = 0; i < N; i++) {
            map[i] = br.readLine().toCharArray();
        }

        int W = 0, B = 0;
        for(int i=0; i<N; i++){
            for(int j=0; j<M; j++){
                char team = map[i][j];
                if(map[i][j]=='.'){ continue;}
                result = 0;
                dfs(i, j, team);
                if(team=='W'){W+=result*result;}
                else{B+=result*result;}
            }
        }
        System.out.println(W+" "+B);
    }
}

 

 

결과