728x90
DFS, BFS문제입니다.
N개의 노드로 이루어진 트리가 주어집니다.
이때 각 노드간의 거리가 주어졌을 때 주어진 두 노드간의 거리값을 구하여야 합니다.
N개의 노드는 하나의 트리이므로 싸이클이 존재하지 않습니다.
하나의 트리이므로 각 노드는 모두 연결되어 있음을 알 수 있습니다.
노드간의 거리를 기록해두고 DFS로 탐색하여 두노드 사이의 거리를 찾아냈습니다.
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.StringTokenizer;
public class Main {
static int n;
static int m;
static int[] visit = new int[1001];
static ArrayList<ArrayList<Pair>> al;
static BufferedReader br;
static BufferedWriter bw;
public static void main(String[] args) throws IOException {
br = new BufferedReader(new InputStreamReader(System.in));
bw = new BufferedWriter(new OutputStreamWriter(System.out));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
m = Integer.parseInt(st.nextToken());
al = new ArrayList<>();
for (int i = 0; i <= n; ++i) {
ArrayList<Pair> temp = new ArrayList<>();
al.add(temp);
}
for (int i = 0; i < n - 1; ++i) {
st = new StringTokenizer(br.readLine());
int a = Integer.parseInt(st.nextToken());
int b = Integer.parseInt(st.nextToken());
int c = Integer.parseInt(st.nextToken());
al.get(a).add(new Pair(b, c));
al.get(b).add(new Pair(a, c));
}
for (int i = 0; i < m; ++i) {
st = new StringTokenizer(br.readLine());
int start = Integer.parseInt(st.nextToken());
int end = Integer.parseInt(st.nextToken());
visit[start] = 1;
DFS(start, end, 0);
visit[start] = 0;
}
bw.close();
}
public static void DFS(int s, int e, int num) throws IOException {
if(s == e) {
bw.write(num + "\n");
return;
}
for(int i=0; i<al.get(s).size(); ++i) {
if(visit[al.get(s).get(i).link] == 1) {
continue;
}
visit[al.get(s).get(i).link] = 1;
DFS(al.get(s).get(i).link, e, num + al.get(s).get(i).value);
visit[al.get(s).get(i).link] = 0;
}
}
public static class Pair {
public int link;
public int value;
public Pair(int link, int value) {
this.link = link;
this.value = value;
}
}
}
'알고리즘' 카테고리의 다른 글
[Java] 프로그래머스 - 징검다리 건너기 (0) | 2023.03.26 |
---|---|
[Java] 백준 2447번 별 찍기 10 (0) | 2023.03.25 |
[Java] 백준 1477번 휴게소 세우기 (1) | 2023.03.20 |
[Java] 백준 17268번 미팅의 저주 (1) | 2023.03.14 |
[Java] 백준 1670번 정상 회담 2 (1) | 2023.03.14 |