[백준] 결혼식 (5567번)

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다. 

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

예제 입력 1 복사

6 5 1 2 1 3 3 4 2 3 4 5

예제 출력 1 복사

3

---

bfs로 모든 친구를 방문해서 풀었다.

친구관계는 친구의 친구까지 되므로, visit배열의 값이 3인 사람까지가 친구이다.

bfs 과정은 그냥 평범하고, vector를 이용해 양방향 그래프로 설정해야 한다.

단방향으로 했다가 틀려서 고쳤다.

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/BaekJoon/2020/%EA%B2%B0%ED%98%BC%EC%8B%9D.cpp

chosh95/STUDY

C++ 코드

#include <iostream>
#include <queue>
#include <algorithm>
#include <vector>
using namespace std;
int N, M;
vector<int> v[502];
int visit[502];

void bfs()
{
	visit[1] = 1;
	queue<int> q;
	q.push(1);
	while (!q.empty()) {
		int cur = q.front();
		q.pop();
		for (int i = 0; i < v[cur].size(); i++) {
			int next = v[cur][i];
			if (visit[next] > visit[cur] + 1) {
				visit[next] = visit[cur] + 1;
				q.push(next);
			}
		}
	}
}

int main()
{
	cin >> N >> M;
	for (int i = 0; i <= N; i++) visit[i] = 5;
	for (int a, b, i = 0; i < M; i++) {
		cin >> a >> b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	bfs();

	int res = 0;
	for (int i = 2; i <= N; i++) {
		if (visit[i] <= 3) res++;
	}
	cout << res;
}