본문 바로가기

CS/알고리즘 풀이

[ 백준 17070 ] 파이프 옮기기 1 ( 자바 )

[ 백준 17070 ] 파이프 옮기기 1 ( java )

유현이가 새 집으로 이사했다. 새 집의 크기는 N×N의 격자판으로 나타낼 수 있고, 1×1크기의 정사각형 칸으로 나누어져 있다. 각각의 칸은 (r, c)로 나타낼 수 있다. 여기서 r은 행의 번호, c는 열의 번호이고, 행과 열의 번호는 1부터 시작한다. 각각의 칸은 빈 칸이거나 벽이다.

오늘은 집 수리를 위해서 파이프 하나를 옮기려고 한다. 파이프는 아래와 같은 형태이고, 2개의 연속된 칸을 차지하는 크기이다.

 


파이프는 회전시킬 수 있으며, 아래와 같이 3가지 방향이 가능하다.


파이프는 매우 무겁기 때문에, 유현이는 파이프를 밀어서 이동시키려고 한다. 벽에는 새로운 벽지를 발랐기 때문에, 파이프가 벽을 긁으면 안 된다. 즉, 파이프는 항상 빈 칸만 차지해야 한다.

파이프를 밀 수 있는 방향은 총 3가지가 있으며, →, ↘, ↓ 방향이다. 파이프는 밀면서 회전시킬 수 있다. 회전은 45도만 회전시킬 수 있으며, 미는 방향은 오른쪽, 아래, 또는 오른쪽 아래 대각선 방향이어야 한다.

파이프가 가로로 놓여진 경우에 가능한 이동 방법은 총 2가지, 세로로 놓여진 경우에는 2가지, 대각선 방향으로 놓여진 경우에는 3가지가 있다.

 

 


아래 그림은 파이프가 놓여진 방향에 따라서 이동할 수 있는 방법을 모두 나타낸 것이고, 꼭 빈 칸이어야 하는 곳은 색으로 표시되어져 있다.


가장 처음에 파이프는 (1, 1)와 (1, 2)를 차지하고 있고, 방향은 가로이다. 파이프의 한쪽 끝을 (N, N)로 이동시키는 방법의 개수를 구해보자.

(최단 거리 문제가 아니다. 이런 가짓수 문제는 dp로도 가능한 경우 존재)

 

 

출처: 백준, https://www.acmicpc.net/problem/17070

 

17070번: 파이프 옮기기 1

유현이가 새 집으로 이사했다. 새 집의 크기는 N×N의 격자판으로 나타낼 수 있고, 1×1크기의 정사각형 칸으로 나누어져 있다. 각각의 칸은 (r, c)로 나타낼 수 있다. 여기서 r은 행의 번호, c는 열의 번호이고, 행과 열의 번호는 1부터 시작한다. 각각의 칸은 빈 칸이거나 벽이다. 오늘은 집 수리를 위해서 파이프 하나를 옮기려고 한다. 파이프는 아래와 같은 형태이고, 2개의 연속된 칸을 차지하는 크기이다. 파이프는 회전시킬 수 있으며, 아래와 같이

www.acmicpc.net

 

 

 

 

핵심 

1 N이 작아 완탐으로 풀 수 있다. ( 또한, dp로도 풀 수 있다. )

 

2 입력 주의 

1) 행열은 1부터 시작

2) dfs 시작은 (1, 2)

3) dfs 종료 조건에서 종료 조건과 end 조건을 주의

 

3 dp로 푼다면? 

dp[y][x][status]로 ( y, x)에서 status일 때, 이 이후로 N, N까지 도달하는데 가짓수 

처음에 -1로 초기화! 시간이 배로 차이 난다.

 

 

 

추가적으로 더 궁금한 점 있으면 댓글 달아주세요

 

 

코드 ( dfs )

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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
 
 
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        map = new int[N+1][N+1];
        
        String[] input;
        for(int i = 1 ; i <= N ; i++) {
            input = br.readLine().split(" ");
            for(int j = 1 ; j <= N ; j++) {
                map[i][j] = Integer.parseInt(input[j-1]);
            }
        }
        
        dfs(120);
        System.out.println(ans);
        
    }
    static int N;
    static int[][] map;
    
    static int ans = 0;
    // 현재 status상태로 놓여 있을 때, 가능한 모든 방향으로 이동한다. 
    static void dfs(int y, int x, int status) {
        
        
        if(!isRange(y, x) || map[y][x] == 1)
            return;
        
        if(y == N && x == N) {
            ans++;
            return;
        }
 
        int ny, nx;
        if(status == 0) {
            ny = y + right[0];
            nx = x + right[1];
            dfs(ny, nx, 0);
            
            boolean flag = true;
            for(int dir = 0 ; dir < 3 ; dir++) {
                ny = y + rdown[dir][0];
                nx = x + rdown[dir][1];
                if(!(isRange(ny, nx) && map[ny][nx] == 0))
                    flag = false;
            }
            if(flag) {
                dfs(y+1, x+12);
            }
        }else if(status == 1) {
            ny = y + down[0];
            nx = x + down[1];
            dfs(ny, nx, 1);
            
            boolean flag = true;
            for(int dir = 0 ; dir < 3 ; dir++) {
                ny = y + rdown[dir][0];
                nx = x + rdown[dir][1];
                if(!(isRange(ny, nx) && map[ny][nx] == 0))
                    flag = false;
            }
            if(flag) {
                dfs(y+1, x+12);
            }
        }else {
            ny = y + right[0];
            nx = x + right[1];
            dfs(ny, nx, 0);
            
            ny = y + down[0];
            nx = x + down[1];
            dfs(ny, nx, 1);
            
            boolean flag = true;
            for(int dir = 0 ; dir < 3 ; dir++) {
                ny = y + rdown[dir][0];
                nx = x + rdown[dir][1];
                if(!(isRange(ny, nx) && map[ny][nx] == 0))
                    flag = false;
            }
            if(flag) {
                dfs(y+1, x+12);
            }
        }
        
    }
    
    static int[] right = {01};
    static int[] down = {10};
    static int[][] rdown = {
            {01},
            {11},
            {10}
    };
    static boolean isRange(int y, int x) {
        return y >= 1 && y <= N && x >= 1 &&  x <= N;
    }
    
}
 
 
cs

 

 

 

코드 ( dp )

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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
 
 
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        map = new int[N+1][N+1];
        dp = new int[N+1][N+1][3];
        
        String[] input;
        for(int i = 1 ; i <= N ; i++) {
            input = br.readLine().split(" ");
            for(int j = 1 ; j <= N ; j++) {
                map[i][j] = Integer.parseInt(input[j-1]);
                dp[i][j][0= -1;
                dp[i][j][1= -1;
                dp[i][j][2= -1;
            }
        }
        
        System.out.println(dfs(120));
        
    }
    static int N;
    static int[][] map;
    
    static int[][][] dp;
    // 현재 status상태로 놓여 있을 때 이후 N, N에 도착할 수 있는 가짓수 구하기  
    static int dfs(int y, int x, int status) {
        if(!isRange(y, x) || map[y][x] == 1)
            return 0;
        
        if(y == N && x == N) {
            return 1;
        }
        
        if(dp[y][x][status] != -1)
            return dp[y][x][status];
        
        int ny, nx, cnt = 0;
        if(status == 0) {
            ny = y + right[0];
            nx = x + right[1];
            cnt += dfs(ny, nx, 0);
            
            boolean flag = true;
            for(int dir = 0 ; dir < 3 ; dir++) {
                ny = y + rdown[dir][0];
                nx = x + rdown[dir][1];
                if(!(isRange(ny, nx) && map[ny][nx] == 0))
                    flag = false;
            }
            if(flag) {
                cnt += dfs(y+1, x+12);
            }
        }else if(status == 1) {
            ny = y + down[0];
            nx = x + down[1];
            cnt += dfs(ny, nx, 1);
            
            boolean flag = true;
            for(int dir = 0 ; dir < 3 ; dir++) {
                ny = y + rdown[dir][0];
                nx = x + rdown[dir][1];
                if(!(isRange(ny, nx) && map[ny][nx] == 0))
                    flag = false;
            }
            if(flag) {
                cnt += dfs(y+1, x+12);
            }
        }else {
            ny = y + right[0];
            nx = x + right[1];
            cnt += dfs(ny, nx, 0);
            
            ny = y + down[0];
            nx = x + down[1];
            cnt += dfs(ny, nx, 1);
            
            boolean flag = true;
            for(int dir = 0 ; dir < 3 ; dir++) {
                ny = y + rdown[dir][0];
                nx = x + rdown[dir][1];
                if(!(isRange(ny, nx) && map[ny][nx] == 0))
                    flag = false;
            }
            if(flag) {
                cnt += dfs(y+1, x+12);
            }
        }
        return dp[y][x][status] = cnt;
    }
    
    static int[] right = {01};
    static int[] down = {10};
    static int[][] rdown = {
            {01},
            {11},
            {10}
    };
    static boolean isRange(int y, int x) {
        return y >= 1 && y <= N && x >= 1 &&  x <= N;
    }
    
}
cs