148 lines
2.8 KiB
Go
148 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
content := returnContent("../input")
|
|
//content := returnContent("../testInput")
|
|
|
|
//Assisted Solution
|
|
|
|
/*struggled quite a bit on this challenge,
|
|
looked at a pre-comepleted solution in order
|
|
to get a valid answer */
|
|
|
|
sum := 0
|
|
|
|
for _, line := range *content {
|
|
split := strings.Split(line, " | ")
|
|
|
|
lhs := split[0]
|
|
rhs := split[1]
|
|
|
|
lhsDigits := strings.Fields(lhs)
|
|
rhsDigits := strings.Fields(rhs)
|
|
|
|
one := ""
|
|
four := ""
|
|
seven := ""
|
|
eight := ""
|
|
var len5 []string
|
|
var len6 []string
|
|
for _, digit := range lhsDigits {
|
|
switch len(digit) {
|
|
case 2:
|
|
one = digit
|
|
case 4:
|
|
four = digit
|
|
case 3:
|
|
seven = digit
|
|
case 7:
|
|
eight = digit
|
|
case 5:
|
|
len5 = append(len5, digit)
|
|
case 6:
|
|
len6 = append(len6, digit)
|
|
}
|
|
}
|
|
|
|
zero := ""
|
|
six := ""
|
|
nine := ""
|
|
for _, digit := range len6 {
|
|
switch inCommon(digit, four) {
|
|
case 4:
|
|
nine = digit
|
|
case 3:
|
|
if inCommon(digit, seven) == 3 {
|
|
zero = digit
|
|
} else {
|
|
six = digit
|
|
}
|
|
}
|
|
}
|
|
|
|
two := ""
|
|
three := ""
|
|
five := ""
|
|
for _, digit := range len5 {
|
|
if inCommon(digit, nine) == 5 {
|
|
if inCommon(digit, seven) == 3 {
|
|
three = digit
|
|
} else {
|
|
five = digit
|
|
}
|
|
} else {
|
|
two = digit
|
|
}
|
|
}
|
|
|
|
var answerDigits [4]int
|
|
for i, digit := range rhsDigits {
|
|
if len(digit) == 6 && inCommon(digit, zero) == 6 {
|
|
answerDigits[i] = 0
|
|
} else if len(digit) == 2 && inCommon(digit, one) == 2 {
|
|
answerDigits[i] = 1
|
|
} else if len(digit) == 5 && inCommon(digit, two) == 5 {
|
|
answerDigits[i] = 2
|
|
} else if len(digit) == 5 && inCommon(digit, three) == 5 {
|
|
answerDigits[i] = 3
|
|
} else if len(digit) == 4 && inCommon(digit, four) == 4 {
|
|
answerDigits[i] = 4
|
|
} else if len(digit) == 5 && inCommon(digit, five) == 5 {
|
|
answerDigits[i] = 5
|
|
} else if len(digit) == 6 && inCommon(digit, six) == 6 {
|
|
answerDigits[i] = 6
|
|
} else if len(digit) == 3 && inCommon(digit, seven) == 3 {
|
|
answerDigits[i] = 7
|
|
} else if len(digit) == 7 && inCommon(digit, eight) == 7 {
|
|
answerDigits[i] = 8
|
|
} else {
|
|
answerDigits[i] = 9
|
|
}
|
|
}
|
|
|
|
sum += 1000*answerDigits[0] + 100*answerDigits[1] + 10*answerDigits[2] + answerDigits[3]
|
|
}
|
|
|
|
fmt.Println(sum)
|
|
|
|
}
|
|
|
|
func inCommon(a string, b string) (count int) {
|
|
for _, m := range a {
|
|
for _, n := range b {
|
|
if m == n {
|
|
count++
|
|
}
|
|
}
|
|
}
|
|
|
|
return count
|
|
}
|
|
|
|
func returnContent(path string) *[]string {
|
|
//read file and return it as an array of integers
|
|
|
|
file, err := os.Open(path)
|
|
var content []string
|
|
|
|
if err != nil {
|
|
fmt.Println("Unlucky, the file didn't open")
|
|
return &content
|
|
}
|
|
defer file.Close()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
content = append(content, scanner.Text())
|
|
}
|
|
|
|
return &content
|
|
}
|