我正在尝试使用优先级队列在 Java 中实现 Prim 算法。
我找不到我的错误。:/我只是认识到队列没有正确排序节点。
图表示例:
0 4 7 5
4 0 2 3
7 2 0 1
5 3 1 0
它总是将节点 4 作为第二个节点。所以它像[node1,node4,node2,node3]而不是[node1,node2,node3,node4]对队列进行排序。我对优先级队列做错了什么?
问候
public class PrimAlgorithm {
private static int[] par; // parent
private static int[] key; // value
private static int sum;
public static void prim(Graph g){
Node[] nodes = g.getNodes();
key = new int[g.getMatrix().length];
par = new int[g.getMatrix().length];
PriorityQueue<Node> queue = new PriorityQueue<Node>(42, new Comparator<Node>(){
public int compare(Node v1, Node v2){
return Integer.valueOf(key[v1.getId()-1]).compareTo(Integer.valueOf(key[v2.getId()-1]));
for (Node n : nodes) {
int x = n.getId()-1;
key[x] = 1000;
par[x] = 0;
queue.add(n);
}
key[0] = 0;
while(!queue.isEmpty()) {
Node n = queue.poll();
List<Node> neighbours = n.getNeighbors();
for (Node m : neighbours){
if ( queue.contains(m) && g.getEdge(n, m).getWeight() !=0 && g.getEdge(n, m).getWeight() < key[m.getId()-1]){
par[m.getId()-1] = n.getId();
key[m.getId()-1] = g.getEdge(n, m).getWeight();
}
}
}
for (int i=0; i < key.length; i++){
sum += key[i];
}
System.out.println("Das Gewicht des minimalen Spannbaumes lautet: " + sum);
System.out.println("Der Spannbaum ergibt sich wie folgt: " );
//fängt ab 1 an sonst, hätten wir immer noch einen Nullknoten
for(int i=0; i <par.length; i++){
System.out.println("Der Vorgänger von Knoten: " + " "+ (i+1) + "-> " + par[i] + " Gewicht "
+ key[i]);
}
}
public static void main(String[] args) {
System.out.println("Prim Algorithmus zu Berechnung des minimalen Spannbaums.");
Graph g = new Graph();
prim(g);
}
}