백준 알고리즘] 5567 - 결혼식(JOI 2010 예선 3번)

시간 제한 : 1초

메모리 제한 : 128MB

문제

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

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

입력

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

출력

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

소스코드

#include <iostream>
using namespace std;
int main(void)
{
	int n, m, x, y, cnt = 0;
	int g[501][501] = { 0, };
	bool check[501] = { false,true, };
	cin >> n >> m;
	for (int i = 0; i < m; i++)
	{
		cin >> x >> y;
		g[x][y] = 1;
	}
	for (int i = 2; i <= n; i++)
		if (g[1][i] || g[i][1])
			for (int j = 1; j <= n; j++)
				if (g[i][j] || g[j][i])
					check[i] = check[j] = true;
	for (int i = 2; i <= n; i++)
		if (check[i])
			cnt++;
	cout << cnt;
}

Tip

그래프 이론이라고 분류는 되어 있으나, 어떤 알고리즘처럼 막 난해한 것은 아니기 때문에 처음 접근할 때부터 겁을 먹을 필요가 없다. 연결 여부만 확인해서 연결된 동기들의 수만 확인하면 된다.