[java 백준] 골드 5/ 1240번 노드 사이의 거리

https://www.acmicpc.net/problem/1240

1240번: 노드사이의 거리

import java.io.*;
import java.util.*;
 
import static java.lang.Math.max;
 
public class Main {
    public static class Node {
        int node;
        int len;
        public Node(int node,int len) {
            this.node = node;
            this.len = len;
        }
 
    }
    public static boolean []visit;
    public static List<Node>list[];
    public static int n,m;
    public static int ans;
    public static int []arr;
 
    public static void  DFS(int node1,int node2,int len) {
 
        if(node1==node2){
            ans=len;
 
        }
        visit[node1]=true;
        for(Node nodes:list[node1]){
            if(!visit[nodes.node]){
                visit[nodes.node]=true;
                DFS(nodes.node,node2,len+nodes.len);
            }
        }
 
 
    }
    public static void main(String[]args) throws IOException {
 
 
        Scanner sc=new Scanner(System.in);
        n=sc.nextInt();
        m=sc.nextInt();
        arr=new int[m];
 
        list=new ArrayList[n+1];
        for(int i=0;i<n+1;i++){
            list[i]=new ArrayList<Node>();
        }
 
        for(int i=0;i<n-1;i++){
            int node1=sc.nextInt();
            int node2=sc.nextInt();
            int len=sc.nextInt();
            list[node1].add(new Node(node2,len));
            list[node2].add(new Node(node1,len));
        }
 
 
        for(int i=0;i<m;i++){
            visit=new boolean[n+1];
            ans=0;
            int node1=sc.nextInt();
            int node2=sc.nextInt();
            DFS(node1,node2,0);
 
            arr[i]=ans;
 
        }
 
        for(int i=0;i<m;i++){
            System.out.println(arr[i]);
        }
 
    }
 
}

 

 

다익스트라로 풀었다. 이게 다익스트라가 맞는지는 모르겠지만..ㅋㅋ 간선의 가중치여서 DFS,다익스트라! 이렇게 생각

했던 것 같다.

if(node1==node2){

<-- 여기서 return을 굳이 안해줘도 됐다. node1==node2일때만 ans에 값이 추가된다는 조건이 있고, visit배열로

조건판단을 또 해주기 때문에 분명 재귀는 끝나기 때문이다.

이게 아마 틀렸습니다의 이유였을듯하다. 탐색을 더 해야할 수도 있는데 갑자기 탐색을 끊어버리니까..!