[JAVA]백준_10026_적록색약

문제링크

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

 

 

 

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 Main_10026 {

	static char grid[][];
	static boolean isVisited[][];
	static int zone,colorZone;
	static int dx[] = {0,0,-1,1}; //상하
	static int dy[] = {-1,1,0,0}; //좌우
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		int N = Integer.parseInt(br.readLine());

		grid = new char[N][N];
		
		for(int i=0;i<N;i++) {
			st = new StringTokenizer(br.readLine());
			String str = st.nextToken();
			for(int j=0;j<N;j++)
				grid[i][j] = str.charAt(j);
		}//grid 입력

		Queue<int []> q = new LinkedList<>();
		
		isVisited = new boolean[N][N];
		
		//적록색약이 ㅇ아닌 사람
		for(int i=0;i<N;i++) {
			for(int j=0;j<N;j++) {
				if(!isVisited[i][j]) {
					q.offer(new int[] {i,j});
					zone++;
				} //방문하지 않은 곳이면 큐에 삽입
				while(!q.isEmpty()) {
					//큐 돌면서 상하좌우 확인하기
					int idx[] = q.poll();
					
					for(int d=0;d<4;d++) {
						int nx = idx[0]+dx[d];
						int ny = idx[1]+dy[d];
						
						if(nx<0 || nx>=N || ny<0 || ny>=N) continue; //범위 밖이면 그냥 지나치키
						if(isVisited[nx][ny]) continue; //방문한 곳이면 그냥 지나치기
						if((grid[nx][ny] == grid[i][j])) {
							//방문 안하고 visited이면
							q.offer(new int[] {nx,ny});
							isVisited[nx][ny] = true;
						}	
					}
				}
			}
		}
		
		isVisited = new boolean[N][N];
		for(int i=0;i<N;i++) {
			for(int j=0;j<N;j++) {
				if(grid[i][j]=='G')
					grid[i][j] = 'R';
			}
		}
		//적록색약인 사람
		for(int i=0;i<N;i++) {
			for(int j=0;j<N;j++) {
				if(!isVisited[i][j]) {
					q.offer(new int[] {i,j});
					colorZone++;
				} //방문하지 않은 곳이면 큐에 삽입
				
				while(!q.isEmpty()) {
					//큐 돌면서 상하좌우 확인하기
					int idx[] = q.poll();
					
					for(int d=0;d<4;d++) {
						int nx = idx[0]+dx[d];
						int ny = idx[1]+dy[d];
						
						if(nx<0 || nx>=N || ny<0 || ny>=N) continue; //범위 밖이면 그냥 지나치키
						if(isVisited[nx][ny]) continue; //방문한 곳이면 그냥 지나치기
						if((grid[nx][ny] == grid[i][j])) {
							//방문 안하고 visited이면
							q.offer(new int[] {nx,ny});
							isVisited[nx][ny] = true;
						}	
					}
				}
			}
		}
		
		System.out.println(zone +" "+colorZone);
	}
}