끄적끄적 코딩
article thumbnail
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;
		}
	}
}

검색 태그