mingssssss

1. 문제

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

2. 코드

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

public class Main {
    
    // 전역 변수 설정
	static char[] c;
	static ArrayList<ArrayList<Integer>> list;
	static int[] dx = { -1, 1, 0, 0 };
	static int[] dy = { 0, 0, -1, 1 };
	static int w;
	static int b;
	static boolean[][] visited;
	static int[] answer;

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());

		list = new ArrayList<>();

		int M = Integer.parseInt(st.nextToken());
		int N = Integer.parseInt(st.nextToken());

		visited = new boolean[N][M];
		answer = new int[2];

		for (int i = 0; i < N; i++) {
			list.add(new ArrayList<>());
		}

		for (int i = 0; i < N; i++) {
			c = br.readLine().toCharArray();
			for (int j = 0; j < M; j++) {
				if (c[j] == 'W') {
					list.get(i).add(j, 1);
				} else {
					list.get(i).add(j, 0);
				}
			}
		}
		// 입력 종료 && list 생성(W == 1, B == 0)
        
        // 1과 0을 따로 탐색
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				if (visited[i][j] == false && list.get(i).get(j) == 1) {
					bfs1(i, j, N, M, list, visited);
					answer[0] += w * w;
				} 
				
				if (visited[i][j] == false && list.get(i).get(j) == 0) {
					bfs2(i, j, N, M, list, visited);
					answer[1] += b * b;
				}
			}
		}

		System.out.println(answer[0] + " " + answer[1]);

//		// 출력
//		for (int i = 0; i < N; i++) {
//			for (int j = 0; j < M; j++) {
//				System.out.printf("%d ", list.get(i).get(j));
//			}
//			System.out.println();
//		}

	}

	public static void bfs1 (int r, int c, int N, int M, ArrayList<ArrayList<Integer>> list, boolean[][] visited) {

		Queue<int[]> q = new LinkedList<>();

		q.add(new int[] { r, c });
		visited[r][c] = true;

		w = 1;
		while (!q.isEmpty()) {

			int now[] = q.poll();

			for (int i = 0; i < 4; i++) {

				int nextX = now[0] + dx[i];
				int nextY = now[1] + dy[i];

				if (nextX < 0 || nextY < 0 || nextX >= N || nextY >= M) {
					continue;
				}

				if (visited[nextX][nextY] == true) {
					continue;
				}

				if (list.get(nextX).get(nextY) == 1) {
					w++;
					visited[nextX][nextY] = true;
					q.add(new int[] { nextX, nextY });
					continue;
				}
			}

		}

	}
	
	public static void bfs2(int r, int c, int N, int M, ArrayList<ArrayList<Integer>> list, boolean[][] visited) {

		Queue<int[]> q = new LinkedList<>();

		q.add(new int[] { r, c });
		visited[r][c] = true;
		
		b = 1;
		
		while (!q.isEmpty()) {

			int now[] = q.poll();

			for (int i = 0; i < 4; i++) {

				int nextX = now[0] + dx[i];
				int nextY = now[1] + dy[i];

				if (nextX < 0 || nextY < 0 || nextX >= N || nextY >= M) {
					continue;
				}

				if (visited[nextX][nextY] == true) {
					continue;
				}

				if (list.get(nextX).get(nextY) == 0) {
					b++;
					visited[nextX][nextY] = true;
					q.add(new int[] { nextX, nextY });
					continue;
				}

			}

		}

	}

}

3. 리뷰

입력을 받아서 W와 B가 몇 개가 인접해있는지 확인한 후

bfs 탐색이 끝난 다음 각각의 값을 제곱해서 더했다.

bfs 하나에서 탐색하려고 했는데, 코드 구현이 복잡해서 W와 B를 따로 탐색했다.

N과 M을 바꿔서 입력 받아 인덱스 에러가 발생했다.

입력받을 때 입력 값을 잘 확인하는 습관을 들여야겠다.