diff --git a/01/main.go b/01/main.go index 1303681..13ddaae 100644 --- a/01/main.go +++ b/01/main.go @@ -45,10 +45,23 @@ func main() { return rightList[i] < rightList[j] }) + simMap := make(map[int]int) totalDistance := 0 for i := 0; i < len(leftList); i++ { + for _, rightItem := range rightList { + if rightItem == leftList[i] { + simMap[leftList[i]]++ + } + } + totalDistance += int(math.Abs(float64(leftList[i] - rightList[i]))) } - fmt.Println(totalDistance) + simScore := 0 + for number, count := range simMap { + simScore += number * count + } + + fmt.Printf("total distance: %d\n", totalDistance) + fmt.Printf("total sim score: %d\n", simScore) }