Add faster Go version of Problem 1