2022 day14 part1

1 files changed, 22 insertions, 1 deletions

diff --git a/src/2022/day14/aoc.h b/src/2022/day14/aoc.h
index 94598b9..d2143f6 100644
--- a/src/2022/day14/aoc.h
+++ b/src/2022/day14/aoc.h
@@ -63,13 +63,20 @@ struct cave {
   int height = 0;
 
   char* space;
-  cave(int w, int h): width(w+1), height(h + 1) {
+  cave(int w, int h): width(w + 1), height(h + 1) {
     space = (char *) malloc(width * height);
     for (int i = 0; i < width * height; i++) {
       *(space + i) = '.';
     }
   }
 
+  cave(cave& c) {
+    width = c.width;
+    height = c.height;
+    space = (char *) malloc(width * height);
+    memcpy(space, c.space, width * height);
+  }
+
   rock::pos to(rock::pos p) {
     return {p.x - rock::t3.minx, p.y}; 
   }
@@ -78,6 +85,20 @@ struct cave {
     return *(space + p.y * width + p.x);
   }
 
+  bool valid(rock::pos p) {
+    return p.x >= 0 && p.x < width && p.y >= 0 && p.y < height;
+  }
+
+  rock::pos drop(rock::pos p) {
+    rock::pos ns[] = {{p.x, p.y + 1}, {p.x - 1, p.y + 1}, {p.x + 1, p.y + 1}};
+    for (int i = 0; i < 3; i++) {
+      if (valid(ns[i]) && get(ns[i]) == '.') {
+        return drop(ns[i]);
+      }
+    }
+    return p;
+  }
+
   void mark(rock& r) {
     for (size_t i = 0; i < r.rs.size() - 1 ; i++) {
       auto r1 = r.rs[i];