feat: day 01, part 2

vicainelli · vicainelli · commit 912ec347f724 · 2024-12-02T00:14:06.000+01:00
diff --git a/2024/ts/src/day-01-historian-hysteria.test.ts b/2024/ts/src/day-01-historian-hysteria.test.ts
@@ -1,14 +1,39 @@
 import { describe, it, expect } from "bun:test";
-import { findDistance } from "./day-01-historian-hysteria";
+import { findDistance, findDistancePart_2 } from "./day-01-historian-hysteria";
+
+const fileTest = Bun.file("./src/day-01-historian-hysteria-input.test.txt");
+const contentTest = await fileTest.text();
 
 const file = Bun.file("./src/day-01-historian-hysteria-input.txt");
 const content = await file.text();
 
-describe("Day 1: Historian Hysteria", async () => {
-  it("should return the correct distance", () => {
-    const result = findDistance(content);
-    const expected = 11;
-    expect(result).toBe(expected);
-    expect(typeof content).toBe("string");
+describe("Day 1: Historian Hysteria", () => {
+  describe("part 1", () => {
+    it("should return the correct distance with test file", () => {
+      const result = findDistance(contentTest);
+      const expected = 11;
+      expect(result).toBe(expected);
+      expect(typeof content).toBe("string");
+    });
+    it("should return the correct distance with real file", () => {
+      const result = findDistance(content);
+      const expected = 2192892;
+      expect(result).toBe(expected);
+      expect(typeof content).toBe("string");
+    });
+  });
+  describe("part 2", () => {
+    it("should return the correct distance with test file", () => {
+      const result = findDistancePart_2(contentTest);
+      const expected = 31;
+      expect(result).toBe(expected);
+      expect(typeof content).toBe("string");
+    });
+    it("should return the correct distance", () => {
+      const result = findDistancePart_2(content);
+      const expected = 22962826;
+      expect(result).toBe(expected);
+      expect(typeof content).toBe("string");
+    });
   });
 });
diff --git a/2024/ts/src/day-01-historian-hysteria.ts b/2024/ts/src/day-01-historian-hysteria.ts
@@ -1,5 +1,5 @@
-export function findDistance(input: string): Number {
-  const [column_a, column_b] = input
+function transformInputIntoArrays(input: string): [never[], never[]] {
+  return input
     .trim()
     .split("\n")
     .map((line) => line.split(/\s+/))
@@ -11,6 +11,10 @@ export function findDistance(input: string): Number {
       },
       [[], []],
     );
+}
+
+export function findDistance(input: string): Number {
+  const [column_a, column_b] = transformInputIntoArrays(input);
 
   const sorted_a = column_a.sort();
   const sorted_b = column_b.sort();
@@ -22,3 +26,11 @@ export function findDistance(input: string): Number {
 
   return sum_of_arr;
 }
+
+export function findDistancePart_2(input: string): Number {
+  const [column_a, column_b] = transformInputIntoArrays(input);
+
+  return column_a
+    .map((y) => column_b.filter((x) => x === y).length * y)
+    .reduce((a, b) => a + b, 0);
+}
