https://www.acmicpc.net/problem/14716

 

14716번: 현수막

혁진이의 생각대로 프로그램을 구현했을 때, 현수막에서 글자의 개수가 몇 개인지 출력하여라.

www.acmicpc.net

 

문제

ANT가 처음 알고리즘 대회를 개최하게 되면서 현수막을 내걸었다.

저번 학기 영상처리 수업을 듣고 배웠던 지식을 최대한 응용 해보고 싶은 혁진이는 이 현수막에서 글자가 몇 개인지 알아보는 프로그램을 만들려 한다.

혁진이는 우선 현수막에서 글자인 부분은 1, 글자가 아닌 부분은 0으로 바꾸는 필터를 적용하여 값을 만드는데 성공했다.

그런데 혁진이는 이 값을 바탕으로 글자인 부분 1이 상, 하, 좌, 우, 대각선으로 인접하여 서로 연결되어 있다면 한 개의 글자라고 생각만 하였다.

혁진이가 필터를 적용하여 만든 값이 입력으로 주어졌을 때, 혁진이의 생각대로 프로그램을 구현하면 글자의 개수가 몇 개인지 출력하여라.

입력

첫 번째 줄에는 현수막의 크기인 M와 N가 주어진다. (1 ≤ M, N ≤ 250)

두 번째 줄부터 M+1 번째 줄까지 현수막의 정보가 1과 0으로 주어지며, 1과 0을 제외한 입력은 주어지지 않는다.

출력

혁진이의 생각대로 프로그램을 구현했을 때, 현수막에서 글자의 개수가 몇 개인지 출력하여라.

 

 

접근방법 및 풀이

- 기본적인 BFS 알고리즘으로 해결했다.

- 대각선에 위치해도 인접한 것으로 보기때문에 탐색 시 상하좌우 외에 4가지 방향을 더 추가한다.

- 인풋배열의 0,0부터 bfs를 진행한다.

- 진행 중 방문한 곳의 인풋배열 값을 0 (글자)로 바꿔준다.

- bfs를 몇번 진행했는지가 곧 답이 된다.

#include <iostream>
#include <queue>
#include <algorithm>
#include <vector>
#define VMAX 250+1
#define INF 987654321

using namespace std;

int n,m;
int b[VMAX][VMAX];
int dx[8] = {-1,-1,-1,0,0,1,1,1};
int dy[8] = {1,0,-1,1,-1,1,0,-1};

void bfs(int sx, int sy){
  queue<pair<int,int>> q;
  int v[VMAX][VMAX]={0};
  q.push({sx,sy});
  v[sx][sy]=1;
  while(!q.empty()){
    int x = q.front().first;
    int y = q.front().second;
    q.pop();
    for(int i=0; i<8; i++){
      int nx = x + dx[i];
      int ny = y + dy[i];
      if(nx >=0 && ny >=0 && nx <n && ny <m){
        if(v[nx][ny]==0 && b[nx][ny]==1){
          v[nx][ny]=1;
          b[nx][ny]=0;
          q.push({nx,ny});
        }
      }
    }
  }
}

int main()
{
  int result=0;
  cin >> n >> m;
  for(int i=0; i<n; i++){
    for(int j=0; j<m; j++){
      cin >> b[i][j];
    }
  }
  
  for(int i=0; i<n; i++){
    for(int j=0; j<m; j++){
      if(b[i][j]==1){
        bfs(i,j);
        result++;
      }
    }
  }

  cout << result;

  return 0;
}

/*
8 19
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 1 0 0 0 1 0 0 0 1 0 1 1 1 1 1 0
0 0 1 0 1 0 0 1 1 0 0 1 0 0 0 1 0 0 0
0 1 0 0 0 1 0 1 0 1 0 1 0 0 0 1 0 0 0
0 1 1 1 1 1 0 1 0 1 0 1 0 0 0 1 0 0 0
0 1 0 0 0 1 0 1 0 0 1 1 0 0 0 1 0 0 0
0 1 0 0 0 1 0 1 0 0 0 1 0 0 0 1 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0

*/

 

느낀점

약간의 BFS 응용이 있어서 재밌고도 쉽게 풀렸다.

+ Recent posts