fixed old code for non-square maps

2 files changed, 5 insertions, 5 deletions

diff --git a/2024/06/day06a.cpp b/2024/06/day06a.cpp
index 69b9340..b229c21 100644
--- a/2024/06/day06a.cpp
+++ b/2024/06/day06a.cpp
@@ -55,7 +55,7 @@ public:
                 cells = new char[M * N];
                 for (int i = 0; i < N; i++)
                         for (int j = 0; j < M; j++)
-                                cells[N*i + j] = j < (int)lines[i].size() ?
+                                cells[M*i + j] = j < (int)lines[i].size() ?
                                     lines[i][j] : out_of_bound;
         }
 
@@ -86,7 +86,7 @@ private:
 
         int coord(pair<int, int> p) {
                 auto [i, j] = p;
-                return i >= N || i < 0 || j >= M || j < 0 ? -1 : N*i + j;
+                return i >= N || i < 0 || j >= M || j < 0 ? -1 : M*i + j;
         }
 };
 
diff --git a/2024/06/day06b.cpp b/2024/06/day06b.cpp
index 2498536..cb5d281 100644
--- a/2024/06/day06b.cpp
+++ b/2024/06/day06b.cpp
@@ -62,9 +62,9 @@ public:
                 visited = new int[M * N];
                 for (int i = 0; i < N; i++) {
                         for (int j = 0; j < M; j++) {
-                                cells[N*i + j] = j < (int)lines[i].size() ?
+                                cells[M*i + j] = j < (int)lines[i].size() ?
                                     lines[i][j] : out_of_bound;
-                                visited[N*i + j] = 0;
+                                visited[M*i + j] = 0;
                         }
                 }
         }
@@ -111,7 +111,7 @@ private:
 
         int coord(pair<int, int> p) {
                 auto [i, j] = p;
-                return i >= N || i < 0 || j >= M || j < 0 ? -1 : N*i + j;
+                return i >= N || i < 0 || j >= M || j < 0 ? -1 : M*i + j;
         }
 
         void set_cell(pair<int, int> p, char c) {