[BOJ] 16946 벽 부수고 이동하기4

Time Lapse :23min 44sec

16946.cpp

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
#include <iostream>
#include <queue>
#include <set>
#include <vector>
using namespace std;
int n, m, g = 1;
char map[1000][1001];
int group[1000][1000];
vector<int> vec;
int dx[4] = {0,1,0,-1};
int dy[4] = {-1,0,1,0};
void dfs(int y, int x) {
int cnt = 1;
group[y][x] = g;
queue<pair<int, int>> q;
q.push({y,x});
while(!q.empty()) {
pair<int,int> p = q.front();
q.pop();
for(int i = 0; i < 4; i++) {
int ny = p.first + dy[i];
int nx = p.second + dx[i];
if(0 <= nx && nx < m && 0 <= ny && ny < n && !group[ny][nx] && map[ny][nx] == '0') {
group[ny][nx] = g;
++cnt;
q.push({ny, nx});
}
}
}
vec.push_back(cnt);
++g;
}
int ans(int y, int x) {
if(map[y][x] == '0') return 0;
int ret = 1;
set<int> s;
for(int i = 0; i < 4; i++) {
int ny = y + dy[i];
int nx = x + dx[i];
if(0 <= nx && nx < m && 0 <= ny && ny < n)
s.insert(group[ny][nx]);
}
for(auto val : s)
ret += vec[val];
return ret % 10;
}
int main() {
scanf("%d %d",&n,&m);
for(int i = 0; i < n; i++)
scanf("%s",&map[i]);
vec.push_back(0);
for(int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
if(map[i][j] == '0' && !group[i][j])
dfs(i,j);
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++)
printf("%d",ans(i,j));
printf("\n");
}
}
Author: Song Hayoung
Link: https://songhayoung.github.io/2020/09/11/PS/BOJ/16946/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.