code.go

package main

import (
	"advent-of-code/helper"
	"fmt"
	"path"
	"runtime"
	"strconv"
	"time"
)

func main() {
	_, f, _, _ := runtime.Caller(0)
	cwd := path.Join(path.Dir(f))

	input := helper.ReadInput(cwd, helper.NewLine)
	iValues := input.Strings()

	// Execute
	start := time.Now()
	result01 := part01(iValues)
	result02 := part02(iValues)
	executionTime := helper.ExecutionTime(time.Since(start))

	fmt.Printf("Solution Part 1: %v\n", result01)
	fmt.Printf("Solution Part 2: %v\n", result02)
	fmt.Printf("Execution time: %s\n", executionTime)

	helper.SaveBenchmarkTime(executionTime, cwd)

	// Testing
	helper.TestResults(
		[]helper.TestingValue{
			helper.TestingValue{Result: result01, Expect: 935419},
			helper.TestingValue{Result: result02, Expect: 49880012},
		},
	)
}

// Task code
func part01(input []string) int {
	for _, firstValue := range input {
		for _, secondValue := range input {
			fv, _ := strconv.Atoi(firstValue)
			sv, _ := strconv.Atoi(secondValue)

			sum := fv + sv

			if 2020 == sum {
				return fv * sv
			}
		}
	}

	return 0
}

func part02(input []string) int {
	for _, firstValue := range input {
		for _, secondValue := range input {
			for _, thirdValue := range input {
				fv, _ := strconv.Atoi(firstValue)
				sv, _ := strconv.Atoi(secondValue)
				tv, _ := strconv.Atoi(thirdValue)

				sum := fv + sv + tv

				if 2020 == sum {
					return fv * sv * tv
				}
			}
		}
	}

	return 0
}