984 wrong answer

Author: aQuaYi

Algorithms/0984.string-without-aaa-or-bbb/string-without-aaa-or-bbb.go

@@ -1,6 +1,27 @@
 package problem0984
 
+import "strings"
+
 func strWithout3a3b(A int, B int) string {
+	var sb strings.Builder
+	sb.Grow(A + B)
+
+	a, b := 'a', 'b'
+	if A < B {
+		A, B = B, A
+		a, b = b, a
+	}
+
+	for A > 0 || B > 0 {
+		for i := 2; i > 0 && A > 0; i-- {
+			sb.WriteRune(a)
+			A--
+		}
+		for i := 2; i > 0 && B > 0; i-- {
+			sb.WriteRune(b)
+			B--
+		}
+	}
 
-	return ""
+	return sb.String()
 }

Algorithms/0984.string-without-aaa-or-bbb/string-without-aaa-or-bbb_test.go

@@ -20,12 +20,24 @@ var tcs = []struct {
 		"abb",
 	},
 
+	{
+		1,
+		3,
+		"bbab",
+	},
+
 	{
 		4,
 		1,
 		"aabaa",
 	},
 
+	{
+		2,
+		5,
+		"bbabbab",
+	},
+
 	// 可以有多个 testcase
 }
 
@@ -46,7 +58,8 @@ func Test_strWithout3a3b(t *testing.T) {
 	ast := assert.New(t)
 
 	for _, tc := range tcs {
-		ast.True(isCorrect(tc.A, tc.B, strWithout3a3b(tc.A, tc.B)))
+		str := strWithout3a3b(tc.A, tc.B)
+		ast.True(isCorrect(tc.A, tc.B, str), "A = %d, B = %d, %s", tc.A, tc.B, str)
 	}
 }