diff --git a/main.go b/main.go index a41a6d6..c1a2920 100644 --- a/main.go +++ b/main.go @@ -3,5 +3,5 @@ package main import "log" func main() { - log.Println(problem1(16, []int{10, 15, 3, 7})) + log.Println(problem2([]int{3, 2, 1})) } diff --git a/problem2.go b/problem2.go new file mode 100644 index 0000000..919e85f --- /dev/null +++ b/problem2.go @@ -0,0 +1,18 @@ +package main + +func problem2(nums []int) []int { + var out []int + for i := range nums { + val := multiplySlice(nums[:i]) * multiplySlice(nums[i+1:]) + out = append(out, val) + } + return out +} + +func multiplySlice(nums []int) int { + out := 1 + for _, num := range nums { + out *= num + } + return out +}