Problem
Problem URL : 적록색약
[1] Answer Code (18. 10. 01)
#include<iostream>
#include<cstring>
using namespace std;
char map[101][101];
int v[101][101];
int n,m;
int dx[] = {0,1,0,-1};
int dy[] = {1,0,-1,0};
void dfs(int x, int y){
v[x][y] = 1;
for(int i=0; i<4; i++){
int nx = x + dx[i];
int ny = y + dy[i];
if(nx >= 0 && nx < n && ny >= 0 && ny < n)
if(!v[nx][ny] && map[nx][ny] == map[x][y])
dfs(nx,ny);
}
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
memset(map,0,sizeof(map));
memset(v,0,sizeof(v));
cin >> n;
for(int i=0; i<n; i++){
string s;
cin >> s;
for(int j=0; j<n; j++)
map[i][j] = s[j];
}
int cnt = 0;
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
if( !v[i][j] ){
cnt ++;
dfs(i,j);
}
}
}
cout << cnt << " ";
memset(v,0,sizeof(v));
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
if( map[i][j] == 'G')
map[i][j] = 'R';
}
}
cnt = 0;
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
if( !v[i][j] ){
cnt ++;
dfs(i,j);
}
}
}
cout << cnt << '\n';
return 0;
}
Review
- DFS.