[백준]2667번: 단지번호붙이기

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

2667번: 단지번호붙이기

 

bfs로 풀었음

코드가 약간 지저분한 느낌이다,,

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;

int n;
int map[25][25]; //지도 
bool visited[25][25]; //방문 체크 
queue<pair<int,int>> q;
int cnt=0; //단지별 집의 개수 count 
vector<int> v; //단지별 집의 개수 저장 

int bfs(int x, int y){	
	visited[x][y]=true;
	q.push(make_pair(x,y));
	cnt++;
	while(!q.empty()){
		int a=q.front().first;
		int b=q.front().second;
		q.pop();
		
		if(a<n-1 && !visited[a+1][b] && map[a+1][b]==1){
			q.push(make_pair(a+1,b));
			visited[a+1][b]=1;
			cnt++;
		}
		if(a>0 && !visited[a-1][b] && map[a-1][b]==1){
			q.push(make_pair(a-1,b));
			visited[a-1][b]=1;
			cnt++;
		}

		if(b<n-1 && !visited[a][b+1] && map[a][b+1]==1){
			q.push(make_pair(a,b+1));
			visited[a][b+1]=1;
			cnt++;
		}
		if(b>0 && !visited[a][b-1] && map[a][b-1]==1){
			q.push(make_pair(a,b-1));
			visited[a][b-1]=1;
			cnt++;
		}

	}
	return cnt;
}

int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		string temp;
		cin>>temp;
		for(int j=0;j<n;j++) map[i][j]=temp[j]-'0';
	}
	
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			if(!visited[i][j]){
				if(map[i][j]==0) visited[i][j]=1;
				else v.push_back(bfs(i,j));
				cnt=0;
			}
		}
	}

	cout<<v.size()<<"\n";
	sort(v.begin(),v.end());
	for(int i=0;i<v.size();i++) cout<<v[i]<<"\n";
	return 0;
}