https://www.acmicpc.net/problem/2667
2667번: 단지번호붙이기
<그림 1>과 같이 정사각형 모양의 지도가 있다. 1은 집이 있는 곳을, 0은 집이 없는 곳을 나타낸다. 철수는 이 지도를 가지고 연결된 집들의 모임인 단지를 정의하고, 단지에 번호를 붙이려 한다. 여기서 연결되었다는 것은 어떤 집이 좌우, 혹은 아래위로 다른 집이 있는 경우를 말한다. 대각선상에 집이 있는 경우는 연결된 것이 아니다. <그림 2>는 <그림 1>을 단지별로 번호를 붙인 것이다. 지도를 입력하여 단지수를 출력하고, 각 단지에 속하는 집의 수
www.acmicpc.net
전형적인 미로문제였다 DFS를 이용하여 문제를 해결하였다.
두가지 방법을 이용해서 문제를 풀어보았다.
방법 1
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | import java.io.*; import java.util.*; public class Main { static int res; public static void main(String[] args) throws IOException { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); int n = Integer.parseInt(br.readLine()); int[][] map = new int[n][n]; boolean[][] visit = new boolean[n][n]; ArrayList<Integer> list = new ArrayList<>(); for (int i = 0; i < n; i++) { String[] in = br.readLine().split(""); for (int j = 0; j < n; j++) { map[i][j] = Integer.parseInt(in[j]); } } for (int i = 0; i < map.length; i++) { for (int j = 0; j < map.length; j++) { if(map[i][j]==1 && !visit[i][j]){ res= fun(i,j,map,visit); list.add(res); } } } Collections.sort(list); System.out.println(list.size()); for(int i:list) System.out.println(i); } static int fun(int x,int y,int[][] map , boolean[][] visit){ if(x>=map.length||y>=map.length||x<0||y<0||map[x][y]==0||visit[x][y]){ return 0; } visit[x][y] = true; return 1+fun(x+1,y,map,visit)+fun(x-1,y,map,visit)+fun(x,y+1,map,visit)+fun(x,y-1,map,visit); } } | cs |
방법 2
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | import java.io.*; import java.util.*; public class Main { static int[] offset = {1, -1, 0, 0}; static int[] offset2 = {0, 0, 1, -1}; static int res; public static void main(String[] args) throws IOException { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); int n = Integer.parseInt(br.readLine()); int[][] map = new int[n][n]; boolean[][] visit = new boolean[n][n]; ArrayList<Integer> list = new ArrayList<>(); for (int i = 0; i < n; i++) { String[] in = br.readLine().split(""); for (int j = 0; j < n; j++) { map[i][j] = Integer.parseInt(in[j]); } } for (int i = 0; i < map.length; i++) { for (int j = 0; j < map.length; j++) { if(map[i][j]==1 && !visit[i][j]){ res = 1; Maze(i,j,map,visit); list.add(res); } } } Collections.sort(list); System.out.println(list.size()); for(int i:list) System.out.println(i); } static int Maze(int x, int y, int[][] map, boolean[][] visit) { visit[x][y] = true; for (int i = 0; i < 4; i++) { int tx = x + offset[i]; int ty = y + offset2[i]; if (tx < map.length && ty < map.length && ty >= 0 && tx >= 0) { if (map[tx][ty] == 1 && !visit[tx][ty]) { Maze(tx, ty, map, visit); res++; } } } return res; } } | cs |
'IT > 알고리즘' 카테고리의 다른 글
백준 1012 유기농배추 (0) | 2020.03.23 |
---|---|
백준 1931 회의실배정 (0) | 2020.03.23 |
백준15652 n과m(4) (0) | 2020.03.19 |
백준1003 피보나치 함수 (동적계획법) (0) | 2020.03.12 |
백준2580 스도쿠 (0) | 2020.03.07 |