Algorithm

[BOJ] 15683. 감시

프로그래민 2020. 5. 15. 01:08
반응형

이문제는 복잡한 구현 문제이다.

문제에 주어진 대로 구현을 하면 되는데, 구현을 함에 있어 재귀와 다양한 함수를 활용하였다. 재귀와 백트래킹을 같이 사용하였는데 2차원 배열의 백트래킹을 할땐 재귀함수 안에서 지역변수로 temp[][] 를 선언하고 원래 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
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
package boj2;
 
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;
 
public class Main_bj_15683_감시 {
    static int[][] map;
    
    static int[] di= {-1,0,1,0};
    static int[] dj= {0,1,0,-1};
    
    static List<Pos> cctvs;
    
    static int R,C;
    
    static int answer;
    
    static class Pos{
        int i,j,sort;
        Pos (int i,int j,int s){
            this.i=i;this.j=j;sort=s;
        }
    }
    
    public static void main(String[] args) throws Exception {
        BufferedReader br= new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;
        st=new StringTokenizer(br.readLine());
        R=Integer.parseInt(st.nextToken());
        C=Integer.parseInt(st.nextToken());
        
        cctvs=new ArrayList<>();
        map=new int[R][C];
        
        for(int i=0;i<R;i++) {
            st=new StringTokenizer(br.readLine());
            for(int j=0;j<C;j++) {
                map[i][j]=Integer.parseInt(st.nextToken());
                if(map[i][j]!=0 && map[i][j]!=6)
                    cctvs.add(new Pos(i, j, map[i][j]));
            }
        }
        
        answer=Integer.MAX_VALUE;
        dfs(0);
        System.out.println(answer);
    }
    
    static void dfs(int index) {
        if(index==cctvs.size()) {
            int count=0;
            for(int i=0;i<R;i++) {
                for(int j=0;j<C;j++) {
                    if(map[i][j]==0)
                        count+=1;
                }
            }
            answer=Math.min(answer, count);
            return;
        }
        
        Pos cctv=cctvs.get(index);
        
        int[][] tempArray=new int[R][C];
        
        if(cctv.sort==1) {
            for(int dir=0;dir<4;dir++) {
                backup(tempArray);
                move(cctv,dir);
                dfs(index+1);
                recover(tempArray);
            }
        }else if(cctv.sort==2) {
            for(int dir=0;dir<2;dir++) {
                backup(tempArray);
                move(cctv,dir);
                dfs(index+1);
                recover(tempArray);
            }
        }else if(cctv.sort==3) {
            for(int dir=0;dir<4;dir++) {
                backup(tempArray);
                move(cctv,dir);
                dfs(index+1);
                recover(tempArray);
            }
        }else if(cctv.sort==4) {
            for(int dir=0;dir<4;dir++) {
                backup(tempArray);
                move(cctv,dir);
                dfs(index+1);
                recover(tempArray);
            }
        }else if(cctv.sort==5) {
            backup(tempArray);
            move(cctv,0);
            dfs(index+1);
            recover(tempArray);
        }
    }
    
    static void backup(int[][] tempArray) {
        for(int i=0;i<R;i++) {
            for(int j=0;j<C;j++
                tempArray[i][j]=map[i][j];
        }
    }
    static void recover(int[][] tempArray) {
        for(int i=0;i<R;i++) {
            for(int j=0;j<C;j++
                map[i][j]=tempArray[i][j];
        }
    }
    
    
    static void move(Pos cctv,int dir) {
        if(cctv.sort==1) {
            change(cctv.i,cctv.j,dir);
        }else if(cctv.sort==2){
            if(dir==0) {
                change(cctv.i,cctv.j,0);change(cctv.i,cctv.j,2);
            }else if(dir==1) {
                change(cctv.i,cctv.j,1);change(cctv.i,cctv.j,3);
            }
        }else if(cctv.sort==3) {
            if(dir==0) {
                change(cctv.i,cctv.j,0);change(cctv.i,cctv.j,1);
            }else if(dir==1) {
                change(cctv.i,cctv.j,1);change(cctv.i,cctv.j,2);
            }else if(dir==2) {
                change(cctv.i,cctv.j,2);change(cctv.i,cctv.j,3);
            }else if(dir==3) {
                change(cctv.i,cctv.j,3);change(cctv.i,cctv.j,0);
            }
        }else if(cctv.sort==4) {
            for(int index=0;index<4;index++) {
                if(dir!=index) {
                    change(cctv.i,cctv.j,index);
                }
            }
        }else if(cctv.sort==5) {
            for(int index=0;index<4;index++) {
                change(cctv.i,cctv.j,index);
            }
        }
    }
 
    static void change(int i,int j,int dir) {
        while(true) {
            i=i+di[dir];
            j=j+dj[dir];
            
            if(i<0 || i>=|| j<0 ||j>=C)    //범위밖
                break;
            
            if(map[i][j]==6)    //cctv또는 벽
                break;
            
            if(1<=map[i][j] && map[i][j]<=5)
                continue;
            
            if(map[i][j]==0 || map[i][j]==-1)
                map[i][j]=-1;
        }
    }
}
 
                                        
반응형

'Algorithm' 카테고리의 다른 글

[BOJ] 1976. 여행 가자  (0) 2020.05.15
[BOJ] 11404. 플로이드  (0) 2020.05.15
[BOJ] 2458. 키 순서  (0) 2020.05.14
[BOJ] 1194. 달이 차오른다, 가자.  (0) 2020.05.14
[Programmers] 비밀지도  (0) 2020.05.14