[BOJ] 16137 : 견우와 직녀
풀이
BFS
아래는 백준님 코드이다.
다시풀어봐야지…내 코드는 마지막에 계쏙 어떤반례로 틀렸다고 나온다.
이 코드는 오작교를 만들 수 있는 모든 절벽(교차X)을 한번씩 오작교를 만들어보고(주기인M을 넣어둠),
BFS탐색으로 최소값을 찾는 방법이다.
각 위치에 오작교를 만들고 탐색이 끝나면 다시 0으로 초기화해준다.
코드
#include <iostream>
#include <queue>
#include <cstring>
#include <tuple>
using namespace std;
int n, m;
int a[10][10];
//dist: x,y,waiting
int dist[10][10][20];
int dx[] = {0,0,1,-1};
int dy[] = {1,-1,0,0};
int bfs() {
memset(dist,-1,sizeof(dist));
queue<tuple<int,int,int>> q;
q.push(make_tuple(0,0,0));
dist[0][0][0] = 0;
while (!q.empty()) {
int x,y,t;
tie(x,y,t) = q.front();
q.pop();
if (a[x][y] >= 2 && t % a[x][y] != 0) {
int nt = (t+1)%a[x][y];
if (dist[x][y][nt] == -1) {
dist[x][y][nt] = dist[x][y][t] + 1;
q.push(make_tuple(x,y,nt));
}
} else {
for (int k=0; k<4; k++) {
int nx = x+dx[k];
int ny = y+dy[k];
if (0 <= nx && nx < n && 0 <= ny && ny < n) {
if (a[x][y] >= 2 && a[nx][ny] >= 2) continue;
if (a[nx][ny] >= 1) {
int nt = (dist[x][y][t]+1)%a[nx][ny];
if (dist[nx][ny][nt] == -1) {
dist[nx][ny][nt] = dist[x][y][t] + 1;
q.push(make_tuple(nx,ny,nt));
}
}
}
}
}
}
return dist[n-1][n-1][0];
}
bool can(int i, int j) {
bool garo = false;
if (j-1 >= 0 && a[i][j-1] == 0) garo = true;
if (j+1 < n && a[i][j+1] == 0) garo = true;
bool sero = false;
if (i-1 >= 0 && a[i-1][j] == 0) sero = true;
if (i+1 < n && a[i+1][j] == 0) sero = true;
return !(garo && sero);
}
int main() {
cin >> n >> m;
for (int i=0; i<n; i++) {
for (int j=0; j<n; j++) {
cin >> a[i][j];
}
}
int ans = -1;
for (int i=0; i<n; i++) {
for (int j=0; j<n; j++) {
if (a[i][j] == 0 && can(i, j)) {
a[i][j] = m;
int now = bfs();
if (now != -1) {
if (ans == -1 || ans > now) {
ans = now;
}
}
a[i][j] = 0;
}
}
}
cout << ans << '\n';
return 0;
}
Written on April 13, 2019