[5567] 결혼식, C++

한줄 후기 : 결혼식은 친구만 불러라

 

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

5567번: 결혼식

BFS를 돌려야 한다.

단 친구의 친구까지만 포함되도록!

 

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
bool visit[502]={false}; 
vector<int> met[502];
void bfs(int start);
int cnt = 0;
int main(){
	int n, m;
	cin >> n >> m;
	for(int i = 0; i<m; i++){
		int u, v;
		cin >> u >> v;
		met[u].push_back(v);
		met[v]. push_back(u);
	}
	bfs(1);
	cout << cnt;
}
void bfs(int start){
	queue<int> q;
	visit[start] = true;
	q.push(start);
	for(int i=0; i<= met[start].size(); i++){
		int now = q.front();
		q.pop();
		for(int j=0; j<met[now].size(); j++){
			int next = met[now][j];
			if(visit[next]==false){
				visit[next] = true;
				q.push(next);
				cnt++;
			}
		}
	}
}