인프런 커뮤니티 질문&답변

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.

테스트케이스 4번, 5번에서 에러가 뜹니다.
어디가 잘 못 된건지 궁금합니다.

import java.io.*;
import java.util.*;

class Node implements Comparable<Node>{
	int v1;
	int v2;
	double c;
	Node(int v1, int v2, double c) {
		this.v1=v1;
		this.v2=v2;
		this.c=c;
	}
	@Override
	public int compareTo(Node o) { //double형은 이렇게 한다.
		if(this.c<o.c) return -1;
		else return 1;
	}
}
public class Main {
	public static int n,m;
	public static int[] unf;
	public static ArrayList<Node> graph = new ArrayList<>();
	public static ArrayList<Integer> x = new ArrayList<>();
	public static ArrayList<Integer> y = new ArrayList<>();
	public static int find(int v) {
		if(v==unf[v]) return v;
		else return unf[v] = find(unf[v]);
	}
	
	public static void union(int a, int b) {
		int fa = find(a);
		int fb = find(b);
		if(fa!=fb) unf[fa] = fb;
	}
    public static void main(String[] argvs) {
        Scanner sc = new Scanner(System.in);
        
        n=sc.nextInt();
        m=sc.nextInt();
        unf = new int[n];
        for(int i=0; i<n; i++) unf[i] = i;
        
        for(int i=0; i<n; i++) {
        	int a=sc.nextInt()-1;
        	int b=sc.nextInt()-1;
        	x.add(a);
        	y.add(b);
        }
        
        for(int i=0; i<n; i++) { //점과 점 사이의 거리를 구하는 구문
        	for(int j=i+1; j<n; j++) {
        		double dis = Math.sqrt((x.get(j)-x.get(i)) *(x.get(j)-x.get(i)) + (y.get(j)-y.get(i)) * (y.get(j)-y.get(i)));
        		graph.add(new Node(i,j,dis));
        	}
        }
        
        for(int i=0; i<m; i++) { //이미 연결되어 있는 점들은 union해준다
        	int a=sc.nextInt();
        	int b=sc.nextInt();
        	union(a-1,b-1); 
        }
        Collections.sort(graph);
        double answer=0;
        
        for(int i=0; i<graph.size(); i++) { //크루스칼
        	int fa = find(graph.get(i).v1);
        	int fb = find(graph.get(i).v2);
        	double cost = graph.get(i).c;
        	
        	if(fa!=fb) {
        		//union(fa, fb);
        		unf[fa] = fb;
        		answer+=cost;
        	}
        }
        
        System.out.format("%.2f", answer); //소수점 출력은 System.out.format으로
    }
}