简体   繁体   中英

Dijkstra's algorithm Java Add on adjacencies

import java.util.PriorityQueue;
import java.util.List;
import java.util.ArrayList;
import java.util.Collections;

public class Dijkstra {

public static void computePaths(Vertex source) {
    source.minDistance = 0.;
    PriorityQueue<Vertex> vertexQueue = new PriorityQueue<Vertex>();
    vertexQueue.add(source);

    while (!vertexQueue.isEmpty()) {
        Vertex u = vertexQueue.poll();

        // Visit each edge exiting u
        for (Edge e : u.adjacencies) {
            Vertex v = e.target;
            double weight = e.weight;
            double distanceThroughU = u.minDistance + weight;
            if (distanceThroughU < v.minDistance) {
                vertexQueue.remove(v);

                v.minDistance = distanceThroughU;
                v.previous = u;
                vertexQueue.add(v);
            }
        }
    }
}

public static List<Vertex> getShortestPathTo(Vertex target) {
    List<Vertex> path = new ArrayList<Vertex>();
    for (Vertex vertex = target; vertex != null; vertex = vertex.previous) {
        path.add(vertex);
    }

    Collections.reverse(path);
    return path;
}

    Vertex A = new Vertex("A");
    Vertex B = new Vertex("B");
    Vertex D = new Vertex("D");
    Vertex J = new Vertex("J");
    Vertex M = new Vertex("M");

    // set the edges and weight
    A.adjacencies = new Edge[]{new Edge(M, 8), new Edge(D, 11)};
    M.adjacencies = new Edge[]{new Edge(A, 8), new Edge(J, 10)};
    D.adjacencies = new Edge[]{new Edge(A, 11), new Edge(J, 5)};
    J.adjacencies = new Edge[]{new Edge(D, 5), new Edge(M, 10)};

    computePaths(J); // run Dijkstra
    System.out.println("Distance to " + A + ": " + A.minDistance);
    List<Vertex> path = getShortestPathTo(A);
    System.out.println("Path: " + path);
}
}

class Vertex implements Comparable<Vertex> {

public final String name;
public Edge[] adjacencies;
public double minDistance = Double.POSITIVE_INFINITY;
public Vertex previous;

public Vertex(String argName) {
    name = argName;
}

public String toString() {
    return name;
}

public int compareTo(Vertex other) {
    return Double.compare(minDistance, other.minDistance);
}

}

class Edge {

public final Vertex target;
public final double weight;

public Edge(Vertex argTarget, double argWeight) {
    target = argTarget;
    weight = argWeight;
}
}

Is there a way to add on to the adjacencies? For example:

A.adjacencies = new Edge[]{new Edge(M, 8)};
A.adjacencies = new Edge[]{new Edge(D, 11)};

Basically I want to check if A.adjacencies has any edge and if it does add the second line as a new edge without overwriting the first line.

Is there a way to do it? If there isn't, is there a way to work around this problem?

Use a Collection instead of arrays. For example ArrayLisy:

A.adjacents = new ArrayList <Edge>();
A.add (new Edge (M, 8));
A.add (new Edge (D, 11));

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM