[Algorithm/Java] 백준 1260번 - DFS와 BFS
https://www.acmicpc.net/problem/1260
📌 문제
문제 유형
- 그래프
문제 설명
그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오.
단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다.
정점 번호는 1번부터 N번까지이다.
입출력 예시
입력
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다.
다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다.
어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.
4 5 1
1 2
1 3
1 4
2 4
3 4
출력
첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.
1 2 4 3
1 2 3 4
🔍 문제 풀이
인접 행렬 생성
인접 행렬이란, 그래프를 2차원 배열
graph[][]
로 표현하여 노드 간 연결 상태를 저장하는 방식이다.
graph[i][j] == 1
→ i번 노드와 j번 노드가 연결되어 있음graph[i][j] == 0
→ 연결되어 있지 않음- 문제에서는 무방향 그래프이므로,
graph[a][b] = 1
뿐만 아니라graph[b][a] = 1
도 함께 설정한다.
초기화 방법 (Java)
int[][] graph = new int[n + 1][n + 1]; // 정점 번호가 1부터 시작하므로 +1
graph[a][b] = 1;
graph[b][a] = 1;
간선 정보 저장 예시
입력:
1 2
1 3
1 4
2 4
3 4
인접 행렬:
| | 1 | 2 | 3 | 4 |
|---|---|---|---|---|
| 1 | 0 | 1 | 1 | 1 |
| 2 | 1 | 0 | 0 | 1 |
| 3 | 1 | 0 | 0 | 1 |
| 4 | 1 | 1 | 1 | 0 |
이제 이 행렬을 기준으로 DFS, BFS 탐색을 수행하면 된다.
DFS 동작 과정
- 현재 노드를 방문
- 연결된 정점 중 번호가 작은 것부터 재귀적으로 방문
- 이미 방문한 노드는 패스
호출 | 방문 여부 (visited[] ) |
출력 |
---|---|---|
dfs(1) | [false, ✅1, false, false, false] | 1 |
dfs(2) | [false, ✅1, ✅2, false, false] | 2 |
dfs(4) | [false, ✅1, ✅2, false, ✅4] | 4 |
dfs(3) | [false, ✅1, ✅2, ✅3, ✅4] | 3 |
결과
1 2 4 3
BFS 동작 과정
- 큐에 시작 노드 삽입 → 방문 처리
- 큐에서 꺼낸 노드의 연결된 노드를 큐에 추가
- 가까운 노드부터 넓게 탐색 (Queue 활용)
Step | 큐 상태 | 현재 노드 | 방문 상태 | 출력 |
---|---|---|---|---|
1 | [1] | 1 | [false, ✅1, false, false, false] | 1 |
2 | [2, 3, 4] | 2 | [false, ✅1, ✅2, ✅3, ✅4] | 2 |
3 | [3, 4] | 3 | 그대로 | 3 |
4 | [4] | 4 | 그대로 | 4 |
결과
1 2 3 4
💻 전체 코드
package org.example;
import java.io.*;
import java.util.*;
public class Main {
static int n, m, v;
static int[][] graph;
static boolean[] visited;
public static void dfs(int v){
visited[v] = true;
System.out.print(v + " ");
for(int i=1; i<=n; i++){
if(graph[v][i] == 1 && !visited[i]){
dfs(i);
}
}
}
public static void bfs(int v){
Deque<Integer> dq = new ArrayDeque<>();
dq.offer(v);
visited[v] = true;
while(!dq.isEmpty()){
int cur = dq.poll();
System.out.print(cur + " ");
for(int i=1; i<=n; i++){
if(!visited[i] && graph[cur][i] == 1){
dq.offer(i);
visited[i] = true;
}
}
}
}
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()); // 노드수
m = Integer.parseInt(st.nextToken()); // 간선수
v = Integer.parseInt(st.nextToken()); // 정점 번호
graph = new int[n+1][n+1];
visited = new boolean[n+1];
for(int i=0; i<m; i++){
st = new StringTokenizer(br.readLine());
int a = Integer.parseInt(st.nextToken());
int b = Integer.parseInt(st.nextToken());
graph[a][b] = 1;
graph[b][a] = 1;
}
dfs(v);
System.out.println("");
visited = new boolean[n+1];
bfs(v);
}
}
댓글남기기