This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
char matrix[404][404];
int visited[404][404][2];
int dp[404][404][2], dx[4] = {-1, 1, 0, 0}, dy[4] = {0, 0, 1, -1}, n, m;
bool valid(int x, int y){
return ((1<=x && x<=n) && (1<=y && y<=m));
}
int count_visited(){
int ret = 0;
for (int i=1;i<=n;i++){
for (int j=1;j<=m;j++) if (visited[i][j][0] || visited[i][j][1]) ret++;
}
return ret;
}
bool dfs(int x, int y, int c){
if (visited[x][y][c]) return visited[x][y][c]==1e9;
visited[x][y][c] = c+1;
bool ret = 1;
int nx1 = x+dx[c], nx2 = y+dy[c];
if (valid(nx1, nx2)){
ret &= dfs(nx1, nx2, c);
}
int d = 2;
if (!c && matrix[x][y]=='Z') d = 3;
if (c && matrix[x][y]=='N') d = 3;
int nx3 = x+dx[d], nx4 = y+dy[d];
if (valid(nx3, nx4)){
if (d==2 && matrix[nx3][nx4]=='N') ret &= dfs(nx3, nx4, 0);
else if (d==3 && matrix[nx3][nx4]=='Z') ret &= dfs(nx3, nx4, 0);
else ret &= dfs(nx3, nx4, 1);
}
visited[x][y][c] = 1e9;
return ret;
}
int main(){
scanf("%d %d", &n, &m);
for (int i=1;i<=n;i++){
scanf("%s", matrix[i]+1);
for (int j=1;j<=m;j++) dp[i][j][0] = dp[i][j][1] = 1e9;
}
for (int i=1;i<=n;i++){
for (int j=1;j<=m;j++){
for (int k=1;k<=n;k++){
for (int l=1;l<=m;l++) visited[k][l][0] = visited[k][l][1] = 0;
}
if (dfs(i, j, 0)) dp[i][j][0] = count_visited()<<1;
for (int k=1;k<=n;k++){
for (int l=1;l<=m;l++) visited[k][l][0] = visited[k][l][1] = 0;
}
if (dfs(i, j, 1)) dp[i][j][1] = count_visited()<<1;
//printf("%d %d: %d %d\n", i, j, dp[i][j][0], dp[i][j][1]);
}
}
for (int i=1;i<=n;i++){
for (int j=1;j<=m;j++){
int ans = min(dp[i][j][0], dp[i][j][1]);
if (ans==1e9) ans = -1;
printf("%d ", ans);
}
printf("\n");
}
return 0;
}
Compilation message (stderr)
sandwich.cpp: In function 'int main()':
sandwich.cpp:43:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
43 | scanf("%d %d", &n, &m);
| ~~~~~^~~~~~~~~~~~~~~~~
sandwich.cpp:45:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
45 | scanf("%s", matrix[i]+1);
| ~~~~~^~~~~~~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |