From 863ff91740be5b6023a80cb400319ba3f142303c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Milan=20Aleksic=CC=81?= <milanaleksic@gmail.com>
Date: Wed, 20 Dec 2023 16:19:48 +0100
Subject: [PATCH] 14-1

---
 src/day14.zig | 109 +++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 94 insertions(+), 15 deletions(-)

diff --git a/src/day14.zig b/src/day14.zig
index 66160d3..d39bd56 100644
--- a/src/day14.zig
+++ b/src/day14.zig
@@ -3,58 +3,137 @@ const util = @import("util.zig");
 const mem = std.mem;
 const print = std.debug.print;
 
+const NodeType = enum {
+    Empty,
+    RoundRock,
+    CubeRock,
+};
+
+const Node = struct {
+    const Self = @This();
+    nodeType: NodeType,
+
+    pub fn toString(self: *Self) []const u8 {
+        return switch (self.nodeType) {
+            NodeType.Empty => " ",
+            NodeType.RoundRock => "O",
+            NodeType.CubeRock => "#",
+        };
+    }
+};
+
 const Data = struct {
     const Self = @This();
     allocator: std.mem.Allocator,
 
-    rows: std.ArrayList(std.ArrayList(i64)),
+    rows: std.ArrayList(std.ArrayList(Node)),
 
     pub fn init(allocator: std.mem.Allocator, list: std.ArrayList([]const u8)) !Self {
+        var rows: std.ArrayList(std.ArrayList(Node)) = std.ArrayList(std.ArrayList(Node)).init(allocator);
         for (list.items) |line| {
-            print("line={s}\n", .{line});
+            // print("line={s}\n", .{line});
+            var row: std.ArrayList(Node) = std.ArrayList(Node).init(allocator);
+            for (line) |c| {
+                const nodeType: NodeType = switch (c) {
+                    '#' => NodeType.CubeRock,
+                    'O' => NodeType.RoundRock,
+                    '.' => NodeType.Empty,
+                    else => @panic("incompatible input detected"),
+                };
+                try row.append(Node{
+                    .nodeType = nodeType,
+                });
+            }
+            try rows.append(row);
         }
         return Self{
-            // store data
+            .rows = rows,
             .allocator = allocator,
         };
     }
 
     pub fn deinit(self: *Self) void {
-        _ = self;
-        // deinit data
+        for (self.rows.items) |row| {
+            row.deinit();
+        }
+        self.rows.deinit();
+    }
+
+    pub fn printMap(self: *Self) void {
+        print("\n", .{});
+        for (0..self.rows.items[0].items.len) |_| {
+            print("*", .{});
+        }
+        print("\n", .{});
+        for (self.rows.items) |row| {
+            for (row.items) |*node| {
+                print("{s}", .{node.toString()});
+            }
+            print("\n", .{});
+        }
     }
 };
 
-pub fn part1(allocator: std.mem.Allocator, list: std.ArrayList([]const u8)) !i64 {
-    var sum: i64 = 0;
+pub fn part1(allocator: std.mem.Allocator, list: std.ArrayList([]const u8)) !usize {
+    var sum: usize = 0;
 
     var data = try Data.init(allocator, list);
     defer data.deinit();
 
-    // access input data...
-    // for (data.rows.items) |rowData| {
+    // data.printMap();
+
+    for (0..data.rows.items.len) |x| {
+        var foundation: usize = 0;
+        for (data.rows.items, 0..) |*row, y| {
+            var cell = row.items[x];
+            switch (cell.nodeType) {
+                NodeType.CubeRock => foundation = y + 1,
+                NodeType.RoundRock => {
+                    if (foundation < y) {
+                        // print("Moving item from {d} to {d}\n", .{ y, foundation });
+                        data.rows.items[foundation].items[x].nodeType = NodeType.RoundRock;
+                        data.rows.items[y].items[x].nodeType = NodeType.Empty;
+                        sum += data.rows.items.len - foundation;
+                    } else {
+                        sum += data.rows.items.len - y;
+                    }
+                    foundation += 1;
+                },
+                else => {},
+            }
+        }
+    }
 
-    // }
+    // data.printMap();
 
     return sum;
 }
 
 test "part 1 test 1" {
     var list = try util.parseToListOfStrings([]const u8,
-        \\...
+        \\O....#....
+        \\O.OO#....#
+        \\.....##...
+        \\OO.#O....O
+        \\.O.....O#.
+        \\O.#..O.#.#
+        \\..O..#O..O
+        \\.......O..
+        \\#....###..
+        \\#OO..#....
     );
     defer list.deinit();
 
-    const testValue: i64 = try part1(std.testing.allocator, list);
-    try std.testing.expectEqual(testValue, -1);
+    const testValue: usize = try part1(std.testing.allocator, list);
+    try std.testing.expectEqual(testValue, 136);
 }
 
 test "part 1 full" {
     var data = try util.openFile(std.testing.allocator, "data/input-14.txt");
     defer data.deinit();
 
-    const testValue: i64 = try part1(std.testing.allocator, data.lines);
-    try std.testing.expectEqual(testValue, -1);
+    const testValue: usize = try part1(std.testing.allocator, data.lines);
+    try std.testing.expectEqual(testValue, 113456);
 }
 
 pub fn part2(allocator: std.mem.Allocator, list: std.ArrayList([]const u8)) !i64 {