984 wrong answer

Author: aQuaYi

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

@@ -17,7 +17,7 @@ func strWithout3a3b(A int, B int) string {
 			sb.WriteRune(a)
 			A--
 		}
-		for i := 2; i > 0 && B > 0; i-- {
+		for i := 2; i > 0 && B > 0 && A-B < 2; i-- {
 			sb.WriteRune(b)
 			B--
 		}

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

@@ -38,6 +38,12 @@ var tcs = []struct {
 		"bbabbab",
 	},
 
+	{
+		2,
+		6,
+		"bbabbabb",
+	},
+
 	// 可以有多个 testcase
 }