diff --git a/src/main/MazeMain.java b/src/main/MazeMain.java index cb417a9..941a797 100644 --- a/src/main/MazeMain.java +++ b/src/main/MazeMain.java @@ -1,4 +1,5 @@ import net.curtlewis.maze.algorithm.BinaryTree; +import net.curtlewis.maze.algorithm.Sidewinder; import net.curtlewis.maze.grid.Cell; import net.curtlewis.maze.grid.Grid; @@ -8,9 +9,9 @@ Cell cell = new Cell(1, 2); Grid grid = new Grid(10, 10); - // System.out.println(grid); - BinaryTree bt = new BinaryTree(grid); - System.out.println(bt.toString()); + // BinaryTree maze = new BinaryTree(grid); + Sidewinder maze = new Sidewinder(grid); + System.out.println(maze.toString()); } } diff --git a/src/main/net/curtlewis/maze/algorithm/Sidewinder.java b/src/main/net/curtlewis/maze/algorithm/Sidewinder.java new file mode 100644 index 0000000..789e133 --- /dev/null +++ b/src/main/net/curtlewis/maze/algorithm/Sidewinder.java @@ -0,0 +1,55 @@ +package net.curtlewis.maze.algorithm; + +import java.util.ArrayList; +import java.util.List; +import java.util.Random; + +import net.curtlewis.maze.grid.Cell; +import net.curtlewis.maze.grid.Grid; + +public class Sidewinder { + + private Grid grid; + + public Sidewinder(Grid grid) { + this.grid = grid; + init(); + } + + private void init() { + + for(int r = 0; r < grid.getNumberOfRows(); r++) { + List run = new ArrayList<>(); + + for(int c = 0; c < grid.getNumberOfColumns(); c++) { + Cell cell = grid.getCellAt(r, c); + run.add(cell); + + boolean atEasternBoundary = (cell.getEast() == null); + boolean atNorthernBoundary = (cell.getNorth() == null); + + int zeroOrOne = (int) Math.round(Math.random()); + boolean shouldCloseOut = atEasternBoundary || (!atNorthernBoundary && zeroOrOne == 0); + + if (shouldCloseOut) { + if (run.size() > 0) { + int randomIndex = new Random().nextInt(run.size()); + Cell member = run.get(randomIndex); + if (member.getNorth() != null) { + member.link(member.getNorth()); + run = new ArrayList<>(); + } + } + } else { + cell.link(cell.getEast()); + } + } + } + } + + public String toString() { + System.out.println("Sidewinder Algorithm"); + return grid.toString(); + } + +}