1 files changed, 9 insertions, 10 deletions

diff --git a/src/ch/epfl/maze/physical/zoo/Hamster.java b/src/ch/epfl/maze/physical/zoo/Hamster.java
index 150880c..527e8ce 100644
--- a/src/ch/epfl/maze/physical/zoo/Hamster.java
+++ b/src/ch/epfl/maze/physical/zoo/Hamster.java
@@ -6,8 +6,8 @@ import ch.epfl.maze.util.Direction;
 import ch.epfl.maze.util.Vector2D;
 
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
+import java.util.Set;
 import java.util.stream.Collectors;
 
 /**
@@ -34,15 +34,14 @@ public class Hamster extends ProbabilisticAnimal {
     /**
      * Discard directions known to lead to dead ends.
      *
-     * @param choices An array of choices
-     * @return An array of smart choices
+     * @param choices A set of choices
+     * @return A set of smart choices
      */
-    private Direction[] excludeDeadPaths(Direction[] choices) {
-        return (new ArrayList<>(Arrays.asList(choices)))
+    private Set<Direction> excludeDeadPaths(Set<Direction> choices) {
+        return choices
                 .stream()
                 .filter(dir -> !this.deadPaths.contains(this.getPosition().addDirectionTo(dir)))
-                .collect(Collectors.toList())
-                .toArray(new Direction[0]);
+                .collect(Collectors.toSet());
     }
 
     /**
@@ -50,9 +49,9 @@ public class Hamster extends ProbabilisticAnimal {
      * it learns during its journey.
      */
     @Override
-    public Direction move(Direction[] choices) {
-        Direction[] smartChoices = this.excludeDeadPaths(choices);
-        if (smartChoices.length == 1) this.deadPaths.add(this.getPosition()); // dead end
+    public Direction move(Set<Direction> choices) {
+        Set<Direction> smartChoices = this.excludeDeadPaths(choices);
+        if (smartChoices.size() == 1) this.deadPaths.add(this.getPosition()); // dead end
         return super.move(smartChoices);
     }