972 wrong answer

Author: aQuaYi

Algorithms/0972.equal-rational-numbers/equal-rational-numbers.go

@@ -9,20 +9,26 @@ func isRationalEqual(S string, T string) bool {
 	is, ns, rs := parse(S)
 	it, nt, rt := parse(T)
 
-	if rs == "" && rt == "" {
-		return is == it && ns == nt
-	}
+	// if rs == "" && rt == "" {
+	// 	return is == it && ns == nt
+	// }
 
 	ns, rs = simplify(ns, rs)
 	nt, rt = simplify(nt, rt)
 
 	s, ns, rs := convert(is, ns, rs)
 	t, nt, rt := convert(it, nt, rt)
 
-	return s == t && ns == nt && rs == rt
+	return s == t &&
+		ns == nt &&
+		rs == rt
 }
 
 func parse(s string) (string, string, string) {
+	if strings.HasSuffix(s, ".") {
+		s = s[:len(s)-1]
+	}
+
 	dot := strings.Index(s, ".")
 	if dot == -1 {
 		return s, "0", ""
@@ -78,6 +84,7 @@ func convert(integer, nonRepeat, repeat string) (int, string, string) {
 	if nonRepeat == "" && repeat == "9" {
 		i++
 		repeat = ""
+		nonRepeat = "0"
 	}
 	return i, nonRepeat, repeat
 }

Algorithms/0972.equal-rational-numbers/equal-rational-numbers_test.go

@@ -13,6 +13,18 @@ var tcs = []struct {
 	ans bool
 }{
 
+	{
+		"1.9(0)",
+		"1.8(9)",
+		true,
+	},
+
+	{
+		"0.9(9)",
+		"1.",
+		true,
+	},
+
 	{
 		"0",
 		"0.",