970 wrong answer

Author: aQuaYi

Algorithms/0970.powerful-integers/powerful-integers.go

@@ -1,6 +1,45 @@
 package problem0970
 
+import (
+	"math"
+	"sort"
+)
+
 func powerfulIntegers(x int, y int, bound int) []int {
+	res := make([]int, 0, 1024)
+
+	logBound := math.Log(float64(bound))
+	logX := math.Log(float64(x))
+	logY := math.Log(float64(y))
+
+	maxX := int(math.Pow(float64(x), math.Round(logBound/logX)))
+	maxY := int(math.Pow(float64(y), math.Round(logBound/logY)))
+
+	for i := maxX; i >= 1; i /= x {
+		for j := maxY; j >= 1; j /= y {
+			if i+j > bound {
+				continue
+			}
+			res = append(res, i+j)
+		}
+	}
+
+	res = removeRepeated(res)
+
+	return res
+}
 
-	return nil
+func removeRepeated(nums []int) []int {
+	sort.Ints(nums)
+	res := make([]int, 1, len(nums))
+	res[0] = nums[0]
+	last := nums[0]
+	for _, n := range nums {
+		if n == last {
+			continue
+		}
+		res = append(res, n)
+		last = n
+	}
+	return res
 }

Algorithms/0970.powerful-integers/powerful-integers_test.go

@@ -1,6 +1,7 @@
 package problem0970
 
 import (
+	"sort"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -14,6 +15,13 @@ var tcs = []struct {
 	ans   []int
 }{
 
+	{
+		2,
+		1,
+		10,
+		[]int{2, 3, 4, 5, 8, 9},
+	},
+
 	{
 		2,
 		3,
@@ -35,7 +43,10 @@ func Test_powerfulIntegers(t *testing.T) {
 	ast := assert.New(t)
 
 	for _, tc := range tcs {
-		ast.Equal(tc.ans, powerfulIntegers(tc.x, tc.y, tc.bound), "输入:%v", tc)
+		sort.Ints(tc.ans)
+		ans := powerfulIntegers(tc.x, tc.y, tc.bound)
+		sort.Ints(ans)
+		ast.Equal(tc.ans, ans, "输入:%v", tc)
 	}
 }