import java.util.*; import java.io.*; public class Main { public static void main(String[] args)throws IOException{ BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out)); StringTokenizer st = new StringTokenizer(br.readLine()); int N = Integer.parseInt(st.nextToken()); int M = Integer.parseInt(st.nextToken()); int[][] mapArr = new int[N][M]; boolean[][] visited = new boolean[N][M]; int[][] areaArr = new int[N][M]; // 해당 위치가 어느 영역에 해당하는지 '영역숫자'를 저장하기 위함. -1일경우 할당안됨. Map areaMap = new HashMap(); // 영역숫자와 영역크기 map for(int i=0; i stack = new ArrayDeque<>(); // DFS나 BFS나 같은 결과 int areaNum = 0; ArrayList idxList = new ArrayList(); // mapArr에서 1이 저장된 부분만 저장 (row*M+col) // 영역숫자-영역넓이 구하기 위함 for(int i=0; i=0 && nRow < N && nCol>=0 && nCol < M && !visited[nRow][nCol] && mapArr[nRow][nCol] == 0){ // 이어진 0인 경우 count++; visited[nRow][nCol]=true; areaArr[nRow][nCol] = areaNum; stack.push(nRow * 1001 + nCol); } } } areaMap.put(areaNum,count); // areaNum와 count map에 더하기 areaNum++; } } } Set set= new HashSet(); for(int i = 0; i=0 && nRow < N && nCol>=0 && nCol < M && areaArr[nRow][nCol]!=-1){ set.add(areaArr[nRow][nCol]); } } int count = 0; for(int area : set){ count+=areaMap.get(area); } set.clear(); mapArr[row][col] = (1 + count)%10; } for(int i=0; i

코드 #6654