[백준] 1647번:도시 분할 계획 (Java 자바)

문제

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

1647번: 도시 분할 계획

 

풀이 및 소스코드

 

MST를 사용해서 최소 가중치로 그래프를 만들어 준 후,

가장 가중치가 큰 간선 하나를 빼면 유지비가 최소가 되는 두 마을을 만들 수 있다.

따라서 원래의 MST는 n-1번의 간선을 만들어줬다면,

이 문제는 n-2번의 간선만 만들어주면 된다.

 

 

• 크루스칼 알고리즘 사용

(320940kb, 1160ms)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {
	public static boolean union(int x, int y) {
		x = find(x);
		y = find(y);
		if(x==y) return false;
		p[x] = y;
		return true;
	}
	public static int find(int x) {
		if(p[x] == x) return x;
		
		return p[x] = find(p[x]);
	}
	static int[] p;
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		st = new StringTokenizer(br.readLine());
		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());
		
		p = new int[n];
		for(int i=0;i<n;i++) {
			p[i] = i;
		}//자기자신을 부모로 초기화
		
		PriorityQueue<node> q = new PriorityQueue<node>();
		
		for(int i=0;i<m;i++) {
			st = new StringTokenizer(br.readLine());
			int a = Integer.parseInt(st.nextToken())-1;
			int b = Integer.parseInt(st.nextToken())-1;
			int w = Integer.parseInt(st.nextToken());
			q.add(new node(a, b, w));
		}
		int cnt = 0;
		int res = 0;
		while(!q.isEmpty()) {
			node now = q.poll();
			if(union(now.x, now.y)) {
				cnt++;
				res += now.w;
			}
			if(cnt==n-2) break;
		}
		System.out.println(res);
	}
}
class node implements Comparable<node>{
	int x,y,w;

	public node(int x, int y,int w) {
		super();
		this.x = x;
		this.y = y;
		this.w = w;
	}

	@Override
	public int compareTo(node o) {
		// TODO Auto-generated method stub
		return this.w-o.w;
	}
	
}