Find all triplets in an array that adds to a zero in Go (Golang)

Table of Contents

Overview

Let’s say the input is

[-3, 0, 3, 4, -1, -6]

Then the answer would be

[[-3 0 3] [-3 4 -1]]

We can use a hash for the solution. It is based upon the idea that

If two of the numbers are x and y

Then the other number will be -(x+y)

Program

package main
import (
"fmt"
)
func main() {
output := threeSumZero([]int{-3, 0, 3, 4, -1, -6})
fmt.Println(output)
}
func threeSumZero(nums []int) [][]int {
numsLength := len(nums)
var results [][]int
for k := 0; k < numsLength-2; k++ {
numsMap := make(map[int]int)
for i := k + 1; i < numsLength; i++ {
if numsMap[0-(nums[i]+nums[k])] > 0 {
result := []int{nums[k], 0 - (nums[i] + nums[k]), nums[i]}
results = append(results, result)
}
numsMap[nums[i]] = i
}
}
return results
}

Output

[[-3 0 3] [-3 4 -1]]

golang