|
| 1 | +package main |
| 2 | + |
| 3 | +import ( |
| 4 | + "fmt" |
| 5 | + "os" |
| 6 | + "strconv" |
| 7 | + "strings" |
| 8 | +) |
| 9 | + |
| 10 | +func main() { |
| 11 | + filenames := []string{"testInput", "input"} |
| 12 | + for _, fileName := range filenames { |
| 13 | + var total int |
| 14 | + fileContents, err := os.ReadFile(fileName) |
| 15 | + if err != nil { |
| 16 | + panic(err) |
| 17 | + } |
| 18 | + fmt.Println(fileName) |
| 19 | + lines := strings.Split(string(fileContents), "\n") |
| 20 | + for _, line := range lines { |
| 21 | + lineParts := strings.Split(line, ":") |
| 22 | + expressionResult, err := strconv.Atoi(lineParts[0]) |
| 23 | + if err != nil { |
| 24 | + panic(err) |
| 25 | + } |
| 26 | + numbersStr := strings.Split(strings.Trim(lineParts[1], " "), " ") |
| 27 | + numbers := make([]int, len(numbersStr)) |
| 28 | + for index, number := range numbersStr { |
| 29 | + numbers[index], err = strconv.Atoi(number) |
| 30 | + if err != nil { |
| 31 | + panic(err) |
| 32 | + } |
| 33 | + } |
| 34 | + possibleResults := getPossibleResults(numbers) |
| 35 | + for _, possibleResult := range possibleResults { |
| 36 | + if possibleResult == expressionResult { |
| 37 | + total += expressionResult |
| 38 | + break |
| 39 | + } |
| 40 | + } |
| 41 | + } |
| 42 | + |
| 43 | + println(total) |
| 44 | + } |
| 45 | + |
| 46 | +} |
| 47 | +func getPossibleResults(numbers []int) []int { |
| 48 | + var err error |
| 49 | + |
| 50 | + if len(numbers) == 2 { |
| 51 | + results := make([]int, 3) |
| 52 | + results[0] = numbers[0] * numbers[1] |
| 53 | + results[1] = numbers[0] + numbers[1] |
| 54 | + results[2], err = strconv.Atoi(strconv.Itoa(numbers[0]) + strconv.Itoa(numbers[1])) |
| 55 | + if err != nil { |
| 56 | + panic(err) |
| 57 | + } |
| 58 | + return results |
| 59 | + } |
| 60 | + possibleResults := getPossibleResults(numbers[:len(numbers)-1]) |
| 61 | + newResults := make([]int, len(possibleResults)*3) |
| 62 | + for index, result := range possibleResults { |
| 63 | + newResults[index] = result * numbers[len(numbers)-1] |
| 64 | + newResults[index+len(possibleResults)] = result + numbers[len(numbers)-1] |
| 65 | + newResults[index+len(possibleResults)*2], err = strconv.Atoi(strconv.Itoa(result) + strconv.Itoa(numbers[len(numbers)-1])) |
| 66 | + if err != nil { |
| 67 | + panic(err) |
| 68 | + } |
| 69 | + } |
| 70 | + return newResults |
| 71 | +} |
0 commit comments