2606번: 바이러스
첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어
www.acmicpc.net
처음에 1번 컴퓨터가 바이러스에 걸리고 이 컴퓨터로 인해 바이러스에 걸리는 컴퓨터가 몇 대인지 출력하는 문제입니다.
이 문제는 간단한 bfs로 해결할 수 있습니다.
입력으로 u와 v가 주어지면 둘이 연결되어있다는 말입니다.
이 그래프는 양방향 그래프이기 때문에 u의 간선에도 v를 추가하고, v의 간선에도 u를 추가해야합니다.
그 다음 1번부터 시작하여 bfs로 연결되어있는 컴퓨터 수를 구하시면 됩니다.
|
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static StringTokenizer st;
static ArrayList<Integer> list[];
static boolean visit[];
static int N, M, ans;
static void bfs() {
Queue<Integer> q = new LinkedList<>();
q.add(1);
visit[1] = true;
while (!q.isEmpty()) {
int v = q.peek();
q.remove();
for (int i = 0; i < list[v].size(); i++) {
int next = list[v].get(i);
if (visit[next])
continue;
q.add(next);
visit[next] = true;
ans++;
}
}
}
static void input() throws Exception {
int u, v;
st = new StringTokenizer(br.readLine());
N = Integer.parseInt(st.nextToken());
list = new ArrayList[N + 1];
visit = new boolean[N + 1];
for (int i = 1; i <= N; i++) {
list[i] = new ArrayList<>();
}
st = new StringTokenizer(br.readLine());
M = Integer.parseInt(st.nextToken());
while (M-- > 0) {
st = new StringTokenizer(br.readLine());
u = Integer.parseInt(st.nextToken());
v = Integer.parseInt(st.nextToken());
list[u].add(v);
list[v].add(u);
}
}
public static void main(String[] args) throws Exception {
input();
bfs();
System.out.println(ans);
}
}
|
cs |
'algorithm > bfs' 카테고리의 다른 글
| boj 14502 연구소 (0) | 2021.02.12 |
|---|---|
| boj 20304 비밀번호 제작 (0) | 2021.02.08 |
| boj 1012 유기농 배추 (0) | 2021.02.03 |
| boj 16956 늑대와 양 (0) | 2021.02.03 |
| boj 20419 화살표 미로 (Easy) (0) | 2021.01.23 |