boj.kr/5567
문제
상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.
상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다.
출력
첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.
풀이과정
첫번째 for : 양방향 그래프로 입력을 받는다.
두번째 for : 1과 친구인 관계를 ch에 1로 저장하고 queue에 친구관계인 node를 push한다.
세번째 while : queue에 들어간 node를 하나씩 꺼내서 그 것과 연결된 노드 중 방문하지 않은 노드의 ch에 2를 저장한다.
마지막 for : ch가 1인 것과 2인 것의 개수를 센다.
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
queue<int> s;
vector<int> a[10001];
int ch[10001] = {0, };
int ans = 0;
int n;
int main() {
cin >> n;
int t;
cin >> t;
for (int i = 0; i < t; ++i) {
int temp1, temp2;
cin >> temp1 >> temp2;
a[temp1].push_back(temp2);
a[temp2].push_back(temp1);
}
int temlen = a[1].size();
for (int i = 0; i < temlen; ++i) {
ch[a[1][i]] = 1;
s.push(a[1][i]);
}
while (!s.empty()) {
int temp = s.front();
s.pop();
int len = a[temp].size();
for (int j = 0; j<len; ++j)
if (ch[a[temp][j]] == 0) {
ch[a[temp][j]] = 2;
}
}
for (int i = 2; i <= n; ++i) {
if (ch[i] == 2 || ch[i] == 1) {
ans++;
}
}
cout << ans;
return 0;
}
'알고리즘' 카테고리의 다른 글
| [DP] 백준 2011 암호코드 (0) | 2018.01.02 |
|---|---|
| [그래프] 백준 1987 알파벳 (0) | 2017.12.14 |
| [EAZY] 백준 1065번 한수 (0) | 2017.08.17 |
| [EAZY] 백준 4673 셀프넘버 (0) | 2017.08.15 |
| 백준 C++ 11721번 (0) | 2017.08.15 |
