[BOJ 2677] 단지번호 붙이기 (2)

#include<iostream>

#include<algorithm>

#include<vector>

using namespace std;

vector<int> v;

int N,cnt;

int small_cnt;

int dx[] = {0,0,1,-1};

int dy[] = {1,-1,0,0};

int arr[30][30] = {0,}; 

int isVisited[30][30] = {0,};

void dfs(int i, int j){

  isVisited[i][j] = 1;

  for(int k=0; k<4; k++){

    int nx = i + dx[k];

    int ny = j + dy[k];

    if(nx>0 && ny>0 && nx<=N && ny<=N){

      if(isVisited[nx][ny]==0 && arr[nx][ny]==1){

        small_cnt++;

        dfs(nx,ny);

      }

    }

  }

}

int main()

{

  scanf("%d",&N);

  for(int i=1; i<=N; i++){

    for(int j=1; j<=N; j++){

      scanf("%1d",&arr[i][j]);

    }

  }

  

  cnt = 0;

  small_cnt = 0;

  for(int i=1; i<=N; i++){

    for(int j=1; j<=N; j++){

      if(isVisited[i][j] == 0 && arr[i][j] == 1){

        small_cnt = 1;

        dfs(i,j);

        v.push_back(small_cnt);

        cnt++; // 반복해서 초기화할 필요가 없음

      }

    }

  }

  printf("%d\n",cnt);

  for(int i=0; i<v.size(); i++)

  {

    printf("%d\n",v[i]);

  }

  return 0;

}