Problem
Problem URL : 연구소
[1] Answer Code (18. 10. 16)
#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#define p pair<int,int>
using namespace std;
// 동서남북
int dx[] = {0,0,1,-1};
int dy[] = {1,-1,0,0};
int n,m;
int map[8][8];
int ans;
vector<p> virus;
vector<p> list;
bool inRange(int x, int y){
if( x < 0 || x >= n || y < 0 || y >= m)
return false;
return true;
}
int cal(){
int v[8][8] = {0,};
queue<p> q;
for(int i=0; i<n; i++){
for(int j=0; j<m; j++){
v[i][j] = map[i][j];
}
}
for(int i=0; i<virus.size(); i++){
q.push(p(virus[i].first, virus[i].second));
}
while (! q.empty()) {
int x = q.front().first;
int y= q.front().second;
q.pop();
for(int i=0; i<4; i++){
int nx = x + dx[i];
int ny = y + dy[i];
if(! inRange(nx, ny)) continue;
if(v[nx][ny] == 2) continue;
if(map[nx][ny] == 1 || map[nx][ny] == 2){
v[nx][ny] = 1;
continue;
}
else if(map[nx][ny] == 0 && v[nx][ny] == 0 ){
v[nx][ny] = 2;
q.push(p(nx,ny));
}
}
}
int cnt = 0;
for(int i=0; i<n; i++){
for(int j=0; j<m; j++){
if(v[i][j] == 0)
cnt ++;
}
}
return cnt;
}
void dfs(int idx, int p_cnt){ // p_cnt is pick count
if(p_cnt == 3){
int res = cal();
ans = ans < res ? res : ans;
return ;
}
int size = (int)list.size();
for(int i=idx; i<size; i++){
int x = list[i].first;
int y = list[i].second;
map[x][y] = 1;
dfs(i+1,p_cnt+1);
map[x][y] = 0;
}
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
ans = -1;
cin >> n >> m;
for(int i=0; i<n; i++){
for(int j=0; j<m; j++){
cin >> map[i][j];
if(map[i][j] == 0)
list.push_back(p(i,j));
else if(map[i][j] == 2)
virus.push_back(p(i,j));
}
}
dfs(0,0);
cout << ans << endl;
return 0;
}
Review
-
삼성 역량 테스트 기출 문제
-
DFS + BFS