فهرست منبع

Not the best solution, but works

Piotr Czajkowski 2 سال پیش
والد
کامیت
387fc6c49a
1فایلهای تغییر یافته به همراه16 افزوده شده و 0 حذف شده
  1. 16 0
      01/code.go

+ 16 - 0
01/code.go

@@ -43,6 +43,21 @@ func part1(input []int) int {
 	return increase
 }
 
+func part2(input []int) []int {
+	var measurements []int
+	index := 0
+	for i, value := range input {
+		if len(input[i:]) > 2 {
+			measurements = append(measurements, value)
+			measurements[index] += input[i+1]
+			measurements[index] += input[i+2]
+			index++
+		}
+	}
+
+	return measurements
+}
+
 func main() {
 	if len(os.Args) < 2 {
 		log.Fatal("Please provide a file name as argument")
@@ -50,4 +65,5 @@ func main() {
 
 	input := readInput(os.Args[1])
 	fmt.Println("Part 1:", part1(input))
+	fmt.Println("Part 2:", part1(part2(input)))
 }