티스토리 뷰
728x90
https://www.acmicpc.net/problem/13023
- 문제 :
BOJ 알고리즘 캠프에는 총 N명이 참가하고 있다. 사람들은 0번부터 N-1번으로 번호가 매겨져 있고, 일부 사람들은 친구이다.
오늘은 다음과 같은 친구 관계를 가진 사람 A, B, C, D, E가 존재하는지 구해보려고 한다.
- A는 B와 친구다.
- B는 C와 친구다.
- C는 D와 친구다.
- D는 E와 친구다.
위와 같은 친구 관계가 존재하는지 안하는지 구하는 프로그램을 작성하시오.
- 풀이 :
DFS문제였습니다.
1. 입력을 받아 그래프를 생성한다.
2. 모든 노드에 대해 dfs를 수행한다. 이 때, 모든 경로를 다 확인하기 위해 아래의 구문을 사용한다.
visited[i] = 1;
dfs(i, 0);
visited[i] = 0;
ex ) 만약 답이 4 -> 3 -> 2 -> 1 -> 0인데 0 -> 1 -> 2 -> 3-> 4 의 순서로 탐색하고 visited를 다시 0으로 갱신하지 않을 경우 0 방문한 노드를 다시 방문하지 않기 때문에 답을 찾을 수 없다.
3. 만약 dfs의 호출 횟수가 4회가 되었을 경우 5개의 노드가 연속되게 존재한다는 것이므로 1을 출력하고 종료한다.
- 소스코드 :
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
//BOJ_13023_ABCDE_G5_208ms
public class Main{
static int N, M, a, b;
static int[] visited;
static boolean end = false;
static Queue<Integer> q;
static ArrayList<Integer>[] graph;
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());
graph = new ArrayList[N];
visited = new int[N];
for (int i = 0; i < N; i++) {
graph[i] = new ArrayList<Integer>();
}
for (int i = 0; i < M; i++) {
st = new StringTokenizer(br.readLine());
a = Integer.parseInt(st.nextToken());
b = Integer.parseInt(st.nextToken());
graph[a].add(b);
graph[b].add(a);
} // 입력
//DFS로 탐색시작
for (int i = 0; i < N; i++) {
visited[i] = 1;
dfs(i, 0);
visited[i] = 0;
}
if(!end) System.out.println(0);
}
private static void dfs(int start, int cnt) {
// 만약 이미 답을 찾았거나, 현재 탐색중인 경로가 정답일 경우 종료
if (end) return ;
if (cnt == 4) {
end = true;
System.out.println(1);
return ;
}
for (int i : graph[start]) {
if(visited[i] == 0) {
visited[i] = 1;
dfs(i, cnt+1);
visited[i] = 0;
}
}
}
}
320x100
'Algorithm > DFS & BFS' 카테고리의 다른 글
(Java/자바) - 백준(BOJ) 13265번 : 색칠하기 (1) | 2022.09.02 |
---|---|
(Java/자바) - 백준(BOJ) 3055번 : 탈출 (0) | 2022.08.24 |
(Java/자바) - SWEA 1238번 :[S/W 문제해결 기본] 10일차 - Contact (0) | 2022.08.23 |
(Java/자바) - 백준(BOJ) 1987번 : 알파벳 (0) | 2022.08.18 |
(Java/자바) - 백준(BOJ) 4963번 : 섬의 개수 (2) | 2022.07.19 |
댓글
© 2022 WonSeok, All rights reserved