[백준] 여행가자 1976 java

오늘내일·2024년 7월 11일
0

최적의 풀이는 아니지만 문제 접근하는 방법이 참고가 될 수 있지 않을까해서 올려봅니다. 코드에 주석 참고해주세요.

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

public class Main {
  // 도시 N개가 있고, 여행계획을 충족시킬 수 있는지 여부 반환
  // 인접리스트를 구성하고
  // 여행 계획의 순서대로 방문해야 하기 때문에, dfs를 통해 여행 계획의 각 도시에서 다음 장소로 이동 가능한지 확인한다.
  static int n, m;
  static List<List<Integer>> list;
  static int[] travelPlan;
  static boolean[] visited;
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    n = Integer.parseInt(br.readLine());
    m = Integer.parseInt(br.readLine());

    // 인접리스트 초기화
    list = new ArrayList<>();
    for (int i = 0; i <= n; i++) {
      list.add(new ArrayList<>());
    }

    // 인접리스트 구성
    StringTokenizer st = null;
    for (int i = 0; i < n; i++) {
      st = new StringTokenizer(br.readLine());
      for (int j = 0; j < n; j++) {
        int canGo = Integer.parseInt(st.nextToken());
        if (canGo == 1) {
          list.get(i + 1).add(j + 1);
        }
      }
    }

    // 여행 계획을 배열로 입력 받는다.
    st = new StringTokenizer(br.readLine());
    travelPlan = new int[m];
    for (int i = 0; i < m; i++) {
      travelPlan[i] = Integer.parseInt(st.nextToken());
    }

    boolean result = true;
    // 여행 계획 배열을 순회하며 각 도시에서 다음 도시로 갈 수 있는 지 확인한다.
    // 다음 도시로 갈 수 없다면 결과는 false로 반복문을 종료한다.
    for (int i = 0; i < m - 1; i++) {
      visited = new boolean[n + 1];
      if (!dfs(travelPlan[i], travelPlan[i + 1])) {
        result = false;
        break;
      }
    }

    if (result) {
      System.out.println("YES");
    } else {
      System.out.println("NO");
    }
  }

  private static boolean dfs(int start, int end) {
    // 탈출조건 : 도착장소와 출발하는 장소가 일치하면 true를 반환한다.
    if (start == end) {
      return true;
    }

    // 중복 방문을 피하기 위해 visit 배열로 방문 도시는 true로 한다.
    visited[start] = true;

    // 출발장소에서 갈 수 있는 도시를 탐색한다.
    List<Integer> cur = list.get(start);
    for (Integer candidate : cur) {
      // 방문한 적 없는 도시만 방문한다.
      if (!visited[candidate]) {
        // 재귀함수 결과가 true면 바로 true를 반환한다.
        if (dfs(candidate, end)) {
          return true;
        }
      }
    }

    // 위 과정에서 true가 반환되지 않았다면 false를 반환한다.
    return false;
  }
}
profile
다시 시작합니다.

0개의 댓글