ACM HDOJ 1875 (畅通工程再续)

题目链接 http://acm.hdu.edu.cn/showproblem.php?pid=1875

程序一 kruskal 算法

import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scn = new Scanner(System.in);
		int casesNumber = Integer.parseInt(scn.next());
		while (0 <= --casesNumber) {
			int pointsNumber = Integer.parseInt(scn.next());
			Kruskal kruskal = new Kruskal(pointsNumber);
			double[] positionX = new double[pointsNumber];
			double[] positionY = new double[pointsNumber];
			for (int i = 0; i < pointsNumber; ++i) {
				positionX[i] = Double.parseDouble(scn.next());
				positionY[i] = Double.parseDouble(scn.next());
			}
			for (int i = 0; i < pointsNumber - 1; ++i) {
				for (int j = i + 1; j < pointsNumber; ++j) {
					double distance = getDistance(positionX[i], positionY[i],
							positionX[j], positionY[j]);
					if (10 <= distance && 1000 >= distance) {
						kruskal.addEdge(i, j, distance);
					}
				}
			}
			double distance = kruskal.calculateDistance() * 100;
			if (kruskal.isConnected()) {
				System.out.println(String.format("%.1f", distance));
			} else {
				System.out.println("oh!");
			}
		}
		scn.close();
	}

	public static double getDistance(double x1, double y1, double x2, double y2) {
		return Math.sqrt(Math.pow(x2 - x1, 2) + Math.pow(y2 - y1, 2));
	}

}

class Kruskal {

	private int[] flag;
	private Queue<Edge> edges;

	private int find(int r) {
		while (r != flag[r]) {
			r = flag[r];
		}
		return r;
	}

	public Kruskal(int pointsNumber) {
		flag = new int[pointsNumber];
		for (int i = 0; i < pointsNumber; ++i) {
			flag[i] = i;
		}
		edges = new PriorityQueue<Edge>();
	}

	public void addEdge(int start, int end, double distance) {
		edges.offer(new Edge(start, end, distance));
	}

	public double calculateDistance() {
		double sumDistance = 0;
		while (!edges.isEmpty()) {
			Edge edge = edges.poll();
			int start = edge.getStart();
			int end = edge.getEnd();
			double distance = edge.getDistance();
			int startRoot = find(start);
			int endRoot = find(end);
			if (startRoot != endRoot) {
				flag[startRoot] = endRoot;
				sumDistance += distance;
			}
		}
		return sumDistance;
	}

	public boolean isConnected() {
		int count = 0;
		for (int i = 0; i < flag.length; ++i) {
			if (i == flag[i]) {
				++count;
				if (2 == count) {
					return false;
				}
			}
		}
		return true;
	}

}

class Edge implements Comparable<Edge> {

	private int start;
	private int end;
	private double distance;

	public Edge(int start, int end, double distance) {
		this.start = start;
		this.end = end;
		this.distance = distance;
	}

	public int getStart() {
		return start;
	}

	public int getEnd() {
		return end;
	}

	public double getDistance() {
		return distance;
	}

	@Override
	public int compareTo(Edge edge) {
		if (this.distance > edge.distance) {
			return 1;
		} else if (this.distance < edge.distance) {
			return -1;
		} else {
			return 0;
		}
	}

}

程序二 prim 算法

import java.util.Arrays;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scn = new Scanner(System.in);
		int casesNumber = Integer.parseInt(scn.next());
		while (0 <= --casesNumber) {
			int pointsNumber = Integer.parseInt(scn.next());
			double[] positionX = new double[pointsNumber];
			double[] positionY = new double[pointsNumber];
			for (int i = 0; i < pointsNumber; ++i) {
				positionX[i] = Double.parseDouble(scn.next());
				positionY[i] = Double.parseDouble(scn.next());
			}
			Prim prim = new Prim(pointsNumber);
			for (int i = 0; i < pointsNumber; ++i) {
				for (int j = i + 1; j < pointsNumber; ++j) {
					double distance = getDistance(positionX[i], positionY[i],
							positionX[j], positionY[j]);
					if (10 <= distance && 1000 >= distance) {
						prim.addEdge(i, j, distance);
					}
				}
			}
			double distance = prim.calculateDistance();
			if (prim.isConnected()) {
				System.out.println(String.format("%.1f", distance * 100));
			} else {
				System.out.println("oh!");
			}
		}
		scn.close();
	}

	public static double getDistance(double x1, double y1, double x2, double y2) {
		return Math.sqrt(Math.pow(x2 - x1, 2) + Math.pow(y2 - y1, 2));
	}

}

class Prim {

	private final double INF = Double.MAX_VALUE / 2;
	private int pointsNumber;
	private double[][] matrix;
	private Edge[] edges;
	private boolean connected;

	public Prim(int pointsNumber) {
		this.pointsNumber = pointsNumber;
		matrix = new double[pointsNumber][pointsNumber];
		for (int i = 0; i < pointsNumber; ++i) {
			Arrays.fill(matrix[i], INF);
			matrix[i][i] = 0;
		}
		edges = new Edge[pointsNumber];
		connected = false;
	}

	public void addEdge(int start, int end, double distance) {
		if (matrix[start][end] > distance) {
			matrix[start][end] = matrix[end][start] = distance;
		}
	}

	public double calculateDistance() {
		double sumDistance = 0;
		for (int i = 0; i < pointsNumber - 1; ++i) {
			edges[i] = new Edge(0, i + 1, matrix[0][i + 1]);
		}
		for (int i = 0; i < pointsNumber - 1; ++i) {
			int minPoint = i;
			double minDistance = edges[i].getDistance();
			for (int j = i + 1; j < pointsNumber - 1; ++j) {
				if (edges[j].getDistance() < minDistance) {
					minDistance = edges[j].getDistance();
					minPoint = j;
				}
			}
			if (INF == minDistance) {
				return INF;
			}
			if (i != minPoint) {
				Edge edge = edges[i];
				edges[i] = edges[minPoint];
				edges[minPoint] = edge;
			}
			sumDistance += edges[i].getDistance();
			for (int j = i + 1; j < pointsNumber - 1; ++j) {
				if (matrix[edges[i].getEnd()][edges[j].getEnd()] < edges[j]
						.getDistance()) {
					edges[j].setDistance(matrix[edges[i].getEnd()][edges[j]
							.getEnd()]);
					edges[j].setStart(edges[i].getEnd());
				}
			}
		}
		connected = true;
		return sumDistance;
	}

	public boolean isConnected() {
		return connected;
	}

}

class Edge {

	private int start;
	private int end;
	private double distance;

	public Edge(int start, int end, double distance) {
		this.start = start;
		this.end = end;
		this.distance = distance;
	}

	public void setStart(int start) {
		this.start = start;
	}

	public int getEnd() {
		return end;
	}

	public double getDistance() {
		return distance;
	}

	public void setDistance(double distance) {
		this.distance = distance;
	}

}