https://www.acmicpc.net/problem/1260
[ 문제 ]
그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.
[ 입력 ]
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.
[ 출력 ]
첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.
[ 입출력 예시 ]
예제 입력 | 예제 출력 |
---|---|
4 5 1 1 2 1 3 1 4 2 4 3 4 | 1 2 4 3 1 2 3 4 |
5 5 3 5 4 5 2 1 2 3 4 3 1 | 3 1 2 5 4 3 1 4 2 5 |
- 정점의 개수(N)을 선언해주고 해주고 값을 입력 받는다.
- 연결된 정점(connected)과 방문 여부(visited)를 표시할 2차원 배열, 1차원 배열을 각각 N+1크기 만큼 선언해준다.
- 간선의 개수 M과 시작 정점 V를 선언하여 각각의 변수에 값을 넣어준다.
이때, M만큼 루프를 돌려 입력 받은 정점1과 정점2 값을 이용해연결된 정점 번호끼리connected[정점1][정점2] = connected[정점2][정점1] = 1
을 넣어 서로 연결된 상태임을 표시한다.[ DFS ]
- 정점의 시작점 v를 매개변수로 하는 dfs 메서드를 선언한다
- 매개변수로 받은 시작점은 간선을 타고 지나온 정점이므로 방문 여부(visited)의 v를 지나왔다고 표시(1)해준다
visited[v] = 1
- 그 후 그 정점을 StringBuilder에 추가하고 정점과 연결된 선을 탐색하면서 방문하지 않았던 정점을 찾아 dfs 메서드를 재귀해준다.
- StringBuilder에 쌓인 값을 출력한다.
- 방문 여부(visited)의 값을 다시 모두 한 번도 방문하지 않은 값들로 초기화하고 StringBuilder()또한 초기화하여 기록할 값들을 초기화해준다.
[BFS]
- 정점의 시작점 v를 매개변수로 하는 bfs 메서드를 선언한다
- 매개변수로 받은 시작점은 간선을 타고 지나온 정점이므로 방문 여부(visited)의 v를 지나왔다고 표시(1)해준다
visited[v] = 1
- 큐를 선언하고 큐에 v값을
offer(v)
해준다.- 큐가 비어있지 않다면(v와 연결된 선을 넣어주고 v를 빼고 그 다음 연결된 값으로 탐색을 할건데 비어있지 않다는 뜻은 계속 연결된 선이 나타난다는 뜻) 첫값(dot)에 q값을
poll()
하여 StringBuilder()에 담고 그 선과 연결되어 있고 방문하지 않은 정점 값을 모두 q에 담아주는 방식을 큐가 비어있을 때까지 반복한다.- StringBuilder() 값을 출력한다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Stack;
import java.util.StringTokenizer;
public class Main {
static int[][] connected;
static int[] visited;
static int N;
static StringBuilder sb = new StringBuilder();
public static void dfs(int v) {
visited[v] = 1;
sb.append(v + " ");
for(int i=1; i<=N; i++) {
if(visited[i] == 0 && connected[v][i] == 1) {
dfs(i);
}
}
}
public static void bfs(int v) {
Queue<Integer> q = new LinkedList<Integer>();
visited[v] = 1;
q.offer(v);
while(!q.isEmpty()) {
int dot = q.poll();
sb.append(Integer.toString(dot)+ " ");
for(int i=1; i<=N; i++) {
// 방문하지 않았고, 연결되어 있다면
if(visited[i] == 0 && connected[dot][i] == 1) {
q.offer(i);
visited[i] = 1;
}
}
}
System.out.println(sb);
}
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 V = Integer.parseInt(st.nextToken());
connected = new int[N+1][N+1];
for(int i=0; i<M; i++) {
st = new StringTokenizer(br.readLine(), " ");
int x = Integer.parseInt(st.nextToken());
int y = Integer.parseInt(st.nextToken());
connected[x][y] = 1;
connected[y][x] = 1;
}
visited = new int[N+1];
dfs(V);
System.out.println(sb);
sb = new StringBuilder();
visited = new int[N+1];
bfs(V);
}
}