update advent-of-code day1

Author: linehk

advent-of-code/2024/day1/day1_test.go

@@ -10,7 +10,7 @@ import (
 	"testing"
 )
 
-func TestDay1(t *testing.T) {
+func TestPart1(t *testing.T) {
 	file, err := os.Open("input")
 	if err != nil {
 		t.Fatal("can't open input file")
@@ -61,3 +61,51 @@ func TestDay1(t *testing.T) {
 
 	fmt.Println(sum)
 }
+
+func TestPart2(t *testing.T) {
+	file, err := os.Open("input")
+	if err != nil {
+		t.Fatal("can't open input file")
+	}
+	defer file.Close()
+
+	var column1 []int
+	var column2 []int
+
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		// ex: 31594   93577
+		line := scanner.Text()
+
+		lineStrSlice := strings.Split(line, "   ")
+		lineIntVar1, err := strconv.Atoi(lineStrSlice[0])
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		lineIntVar2, err := strconv.Atoi(lineStrSlice[1])
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		column1 = append(column1, lineIntVar1)
+		column2 = append(column2, lineIntVar2)
+	}
+
+	if scanner.Err() != nil {
+		t.Fatal("scanner err")
+	}
+
+	var score int
+	for _, v1 := range column1 {
+		var sum int
+		for _, v2 := range column2 {
+			if v1 == v2 {
+				sum += v1
+			}
+		}
+		score += sum
+	}
+
+	fmt.Println(score)
+}

advent-of-code/2024/day1/small_input

@@ -0,0 +1,6 @@
+3   4
+4   3
+2   5
+1   3
+3   9
+3   3