백준
그래프 문제(bfs,dfs) #24479: 알고리즘 수업 - 깊이 우선 탐색 1 [JAVA]
chaechaepower
2024. 2. 18. 17:04
https://www.acmicpc.net/problem/24479
24479번: 알고리즘 수업 - 깊이 우선 탐색 1
첫째 줄에 정점의 수 N (5 ≤ N ≤ 100,000), 간선의 수 M (1 ≤ M ≤ 200,000), 시작 정점 R (1 ≤ R ≤ N)이 주어진다. 다음 M개 줄에 간선 정보 u v가 주어지며 정점 u와 정점 v의 가중치 1인 양
www.acmicpc.net
문제 접근
보통 인접 행렬(2차원 배열)로 그래프를 표현했는데 이 문제는 정점의 수와 간선의 수가 많아 인접 리스트(arraylist)로 표현함.
-> 정점의 수는 엄청 큰데 간선의 수가 엄청 작은 경우 배열 쓰면 메모리 낭비 ㅜㅜ
코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
public class Main {
static List<Integer>[] list;
static boolean[] visited;
static int[] order;
static int cnt;
static int n;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
int m = Integer.parseInt(st.nextToken());
int r = Integer.parseInt(st.nextToken());
order = new int[n + 1];
visited = new boolean[n + 1];
list = new ArrayList[n + 1];
for (int i = 1; i <= n; i++) {
list[i] = new ArrayList<>();
}
for (int i = 0; i < m; i++) {
st = new StringTokenizer(br.readLine());
int a = Integer.parseInt(st.nextToken());
int b = Integer.parseInt(st.nextToken());
list[a].add(b);
list[b].add(a);
}
// 오름차순 정렬
for (int i = 1; i <= n; i++) {
Collections.sort(list[i]);
}
dfs(r);
for (int i = 1; i <= n; i++) {
System.out.println(order[i]);
}
}
static void dfs(int v) {
cnt++;
order[v] = cnt;
visited[v] = true;
List<Integer> l = list[v];
for (int i = 0; i < l.size(); i++) {
int x = l.get(i);
if (!visited[x]) {
dfs(x);
}
}
}
}