diff --git a/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/AStarAlgorithm.java b/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/AStarAlgorithm.java index fd172f0..1345d4e 100644 --- a/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/AStarAlgorithm.java +++ b/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/AStarAlgorithm.java @@ -1,9 +1,17 @@ package org.insa.graphs.algorithm.shortestpath; +import org.insa.graphs.model.Node; + public class AStarAlgorithm extends DijkstraAlgorithm { public AStarAlgorithm(ShortestPathData data) { super(data); } + @Override + protected Label createLabel(Node node) { + return new LabelStar(node, getInputData().getDestination()); + } + + } diff --git a/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/DijkstraAlgorithm.java b/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/DijkstraAlgorithm.java index ad56d94..df2a646 100644 --- a/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/DijkstraAlgorithm.java +++ b/be-graphes-algos/src/main/java/org/insa/graphs/algorithm/shortestpath/DijkstraAlgorithm.java @@ -7,6 +7,7 @@ import org.insa.graphs.algorithm.utils.BinaryHeap; import org.insa.graphs.model.Arc; import org.insa.graphs.model.Graph; import org.insa.graphs.model.Path; +import org.insa.graphs.model.Node; public class DijkstraAlgorithm extends ShortestPathAlgorithm { @@ -14,6 +15,10 @@ public class DijkstraAlgorithm extends ShortestPathAlgorithm { super(data); } + protected Label createLabel(Node node) { + return new Label(node); + } + @Override protected ShortestPathSolution doRun() { @@ -30,7 +35,7 @@ public class DijkstraAlgorithm extends ShortestPathAlgorithm { // Initialize array of labels. ArrayList