/* Your code here! */
#include <vector>
#include "cs225/PNG.h"
#include "dsets.h"
#include <set>

class SquareMaze {
private:
    std::vector<std::vector<bool>*> *bottomWall;
    std::vector<std::vector<bool>*> *rightWall;
    int width;
    int height;
public:
    SquareMaze();
    void makeMaze(int width, int height);
    bool canTravel(int x, int y, int dir) const;
    void setWall(int x, int y, int dir, bool exists);
    std::vector<int> solveMaze();
    cs225::PNG* drawMaze() const;
    cs225::PNG* drawMazeWithSolution();
};