Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | 7 |
8 | 9 | 10 | 11 | 12 | 13 | 14 |
15 | 16 | 17 | 18 | 19 | 20 | 21 |
22 | 23 | 24 | 25 | 26 | 27 | 28 |
29 | 30 | 31 |
Tags
- Kotlin
- 문자열
- 그리디
- Effective Java
- swea
- dfs
- 후니의 쉽게 쓴 시스코 네트워킹
- 에라토스테네스의 체
- 유니온 파인드
- Network
- JUnit 5
- 프로그래머스
- 플로이드-와샬
- mst
- 수학
- 세그먼트 트리
- 스택
- 백트래킹
- 동적계획법
- 시뮬레이션
- java
- CS
- BFS
- 이분탐색
- 알고리즘
- 투 포인터
- 완전탐색
- 위상정렬
- 구현
- 백준
Archives
반갑습니다!
[백준] 5567 결혼식 본문
풀이
상근이의 번호가 1번이므로 1번에서 시작해서 BFS로 친구의 친구까지 확인해주면 된다.
코드
C++
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
int n, m;
vector<vector<int>> adj;
int bfs() {
queue<int> q;
vector<bool> visit(n + 1, false);
q.push(1);
visit[1] = true;
int answer = 0;
int count = 0;
while (!q.empty()) {
int size = q.size();
if (count == 2) break;
while (size--) {
int current = q.front();
q.pop();
for (int i : adj[current]) {
if (!visit[i]) {
visit[i] = true;
q.push(i);
answer++;
}
}
}
count++;
}
return answer;
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> n;
cin >> m;
adj = vector<vector<int>>(n + 1);
for (int i = 0; i < m; i++) {
int a, b;
cin >> a >> b;
adj[a].push_back(b);
adj[b].push_back(a);
}
cout << bfs() << '\n';
return 0;
}
Python3
import sys
from collections import deque
n = int(sys.stdin.readline().rstrip())
m = int(sys.stdin.readline().rstrip())
adj = [[] for _ in range(n + 1)]
for _ in range(m):
a, b = map(int, sys.stdin.readline().rstrip().split())
adj[a].append(b)
adj[b].append(a)
def bfs():
q = deque([1])
visit = [False] * (n + 1)
visit[1] = True
answer = 0
count = 0
while q:
size = len(q)
if count == 2:
break
while size:
current = q.popleft()
for i in adj[current]:
if not visit[i]:
visit[i] = True
q.append(i)
answer += 1
size -= 1
count += 1
return answer
print(bfs())
'알고리즘 문제 풀이' 카테고리의 다른 글
[백준] 18353 병사 배치하기 (0) | 2020.09.25 |
---|---|
[백준] 2491 수열 (0) | 2020.09.25 |
[백준] 1939 중량제한 (0) | 2020.09.24 |
[백준] 1822 차집합 (0) | 2020.09.23 |
[백준] 1715 카드 정렬하기 (0) | 2020.09.23 |