백준 10026 - 적록색약

포인터 클래스 하나 만들고

BFS 두개 로 해결

 

3번 틀렸는데.. 처음에는 런타임 에러 두번 메모리 초과 1번 났음.

 

몇번 수정 끝에 아래처럼 하여 합격 함.

 

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
 
public class Main {
 
    static int[] dx = { -1, 1, 0, 0 };
    static int[] dy = { 0, 0, -1, 1 };
    static boolean visit[][];
    static Queue<Point> q;
    static char map[][];
    static int okCnt,noCnt;    
    static int mapCopy[][];
            
    public static void main(String[] args) throws Exception {
 
        Scanner sc = new Scanner(System.in);
 
        int N = sc.nextInt();
 
        map = new char[N][N];
        mapCopy = new int[N][N];
 
        for (int i = 0; i < map.length; i++) {
            String str = sc.next();
            for (int j = 0; j < str.length(); j++) {
                map[i][j] = str.charAt(j);
            }
        }
    
        //방문, 큐 초기화
        visit = new boolean[N][N];
        q = new LinkedList<>();
        okCnt = 0;
        // 정상적
        for(int i = 0; i<map.length; i++) {
            for(int j = 0 ; j<map[i].length; j++) {
                if(visit[i][j] == false) {
                    okCnt++;
                    BFS(i, j);    
                }
            }
        }
        
        System.out.print(okCnt+" ");
        
        //방문, 큐 초기화
        visit = new boolean[N][N];
        q = new LinkedList<>();
        noCnt = 0;
        
        // 적록이상
        for(int i = 0; i<map.length; i++) {
            for(int j = 0 ; j<map[i].length; j++) {
                if(visit[i][j] == false) {
                    noCnt++;
                    BFS1(i, j);    
                }
            }
        }
        
        System.out.println(noCnt);
        
    }
 
    private static void BFS(int x, int y) {
//        System.out.println(x + " : " + y);
         q.add(new Point(x,y));
         visit[x][y]= true; 
        
         while (!q.isEmpty()) {
             Point p = q.remove();
             //visit[p.x][p.y]= true; 
             
             for(int i = 0; i < 4; i++) {
                 int xx = p.x + dx[i];
                 int yy = p.y + dy[i];
                 // 색맹이 아닐때 구역이 어딘지 확인 가능.
                 // mapCopy[p.x][p.y] = okCnt;
                 
                 if(xx >= 0 && yy >= 0 && xx < map.length && yy < map[0].length && map[xx][yy] == map[p.x][p.y] && !visit[xx][yy]) {
                     visit[xx][yy]= true; 
                     q.add(new Point(xx,yy));
                 }
                 
                 
             }
         }
    }
 
    private static void BFS1(int x, int y) {
//        System.out.println(x + " : " + y);
        q.add(new Point(x,y));
        visit[x][y] = true;
        while (!q.isEmpty()) {
               Point p = q.remove();
               //visit[p.x][p.y] = true;
               
               for(int i = 0 ; i<4; i++) {
                   int xx = p.x + dx[i];
                   int yy = p.y + dy[i];
                   // 색맹일때 구역이 어딘지 확인 가능.
                   // mapCopy[p.x][p.y] = noCnt;
                   if(xx >= 0 && yy >= 0 && xx < map.length && yy < map[0].length && !visit[xx][yy]) {
                       if(map[xx][yy] == map[p.x][p.y] || (map[xx][yy] == 'R' && map[p.x][p.y]== 'G') || (map[xx][yy] == 'G' && map[p.x][p.y]== 'R')) {
                           visit[xx][yy]= true; 
                        q.add(new Point(xx,yy));        
                       }
                }
               }
        }
    }
    
    public static class Point{
        int x = 0;
        int y = 0;
        Point(int _x, int _y){
            x = _x;
            y = _y;
        }
        
    }
}