972 wrong answer

Author: aQuaYi

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

@@ -25,7 +25,7 @@ func isRationalEqual(S string, T string) bool {
 func parse(s string) (string, string, string) {
 	dot := strings.Index(s, ".")
 	if dot == -1 {
-		return s, "", ""
+		return s, "0", ""
 	}
 
 	integer, fraction := s[:dot], s[dot+1:]
@@ -38,6 +38,14 @@ func parse(s string) (string, string, string) {
 	nonRepeat := fraction[:l]
 	repeat := fraction[l+1 : len(fraction)-1]
 
+	if repeat == "0" {
+		repeat = ""
+	}
+
+	if repeat == "" && nonRepeat == "" {
+		nonRepeat = "0"
+	}
+
 	return integer, nonRepeat, repeat
 }
 
@@ -62,10 +70,6 @@ func simplify(nonRepeat, repeat string) (string, string) {
 		}
 	}
 
-	if repeat == "0" {
-		repeat = ""
-	}
-
 	return nonRepeat, repeat
 }
 

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

@@ -14,8 +14,8 @@ var tcs = []struct {
 }{
 
 	{
-		"1.0",
-		"1",
+		"0",
+		"0.",
 		true,
 	},
 
@@ -25,6 +25,12 @@ var tcs = []struct {
 		true,
 	},
 
+	{
+		"1.0",
+		"1",
+		true,
+	},
+
 	{
 		"0.(52)",
 		"0.5(25)",