13023번 ABCDE (Java)

문제 설명

BOJ 알고리즘 캠프에는 총 N명이 참가하고 있다. 사람들은 0번부터 N-1번으로 번호가 매겨져 있고, 일부 사람들은 친구이다.

오늘은 다음과 같은 친구 관계를 가진 사람 A, B, C, D, E가 존재하는지 구해보려고 한다.

• A는 B와 친구다.
• B는 C와 친구다.
• C는 D와 친구다.
• D는 E와 친구다.

위와 같은 친구 관계가 존재하는지 안하는지 구하는 프로그램을 작성하시오.

제한 조건

• 시간제한 : 2초
• 메모리제한 : 512MB

입력 설명

첫째 줄에 사람의 수 N (5 ≤ N ≤ 2000)과 친구 관계의 수 M (1 ≤ M ≤ 2000)이 주어진다.

둘째 줄부터 M개의 줄에는 정수 a와 b가 주어지며, a와 b가 친구라는 뜻이다. (0 ≤ a, b ≤ N-1, a ≠ b) 같은 친구 관계가 두 번 이상 주어지는 경우는 없다.

출력 설명

문제의 조건에 맞는 A, B, C, D, E가 존재하면 1을 없으면 0을 출력한다.

예제 입력 및 출력1

5 4
0 1
1 2
2 3
3 4

---

1

예제 입력 및 출력2

5 5
0 1
1 2
2 3
3 0
1 4

---

1

예제 입력 및 출력3

6 5
0 1
0 2
0 3
0 4
0 5

---

0

예제 입력 및 출력4

8 8
1 7
3 7
4 7
3 4
4 6
3 5
0 4
2 7

---

1

나의 풀이

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.StringTokenizer;


public class Main {
    static ArrayList<Integer>[] arrList;
    static boolean visited[];
    static boolean isAnswer;
    public static void main(String[] args) throws IOException {
        isAnswer = false;
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());  
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        
        arrList = new ArrayList[n];  // 노드 별 엣지 저장
        visited = new boolean[n];    // 노드 별 방문 여부
        
        for(int i=0; i < n; i++){
            arrList[i] = new ArrayList<Integer>();
        }
        
        for(int i=0; i<m; i++){
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());            
            arrList[start].add(end);
            arrList[end].add(start);
        }
        
        for(int i=0; i<n; i++){
            dfs(i, 1);    // 1 depth 부터 dfs 시작
            if(isAnswer) break;
        }
        
        if(isAnswer){
            System.out.println("1");
        } else {
            System.out.println("0");
        }
        
    }
    
    static void dfs(int node, int depth){
        if(depth == 5 || isAnswer){
            // 재귀 횟수가 5번이면 정답 (A-B, B-C, C-D, D-E)
            isAnswer = true;
            return;
        }
        
        visited[node] = true;    // 해당 노드에 방문함
        
        for(int i : arrList[node]){
            if(!visited[i]){
                // 다음 노드를 방문하며 depth를 1씩 증가
                dfs(i, depth + 1);
            }
        }
        
        visited[node] = false;
    }
}

DFS 적용 문제. 문제 설명에 A는 B와 친구, B는 C와 친구, C는 D와 친구, D는 E와 친구인것만 확인하면 되기때문에

주어지는 사람들 (노드, 1~N까지) 중 엣지로 연결되어있는 노드가 5개 이상이면 된다.