972 refract

Author: aQuaYi

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

@@ -14,8 +14,19 @@ func isRationalEqual(S string, T string) bool {
 	tn, tr = simplify(tn, tr)
 	t, tr := convert(ti, tn, tr)
 
-	return s == t &&
-		sr == tr
+	return s == t && sr == tr
+}
+
+// 1.2(3) -> 0.00012(3) -> 00012(3)
+func normalize(s string) string {
+	if !strings.Contains(s, ".") {
+		s += "."
+	}
+	dot := strings.Index(s, ".")
+	if dot < 4 {
+		s = strings.Repeat("0", 4-dot) + s
+	}
+	return strings.Replace(s, ".", "", 1)
 }
 
 func parse(s string) (string, string, string) {
@@ -33,7 +44,7 @@ func parse(s string) (string, string, string) {
 	l := strings.Index(fraction, "(")
 	if l == -1 {
 		if fraction == "0" {
-			fraction = ""
+			return integer, "", ""
 		}
 		return integer, fraction, ""
 	}

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

@@ -153,3 +153,42 @@ func Test_parse(t *testing.T) {
 		})
 	}
 }
+
+func Test_normalize(t *testing.T) {
+	type args struct {
+		s string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+
+		{
+			"1",
+			args{"1"},
+			"0001",
+		},
+
+		{
+			"1234.",
+			args{"1234."},
+			"1234",
+		},
+
+		{
+			"1234.(1)",
+			args{"1234.(1)"},
+			"1234(1)",
+		},
+
+		// Add test cases.
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := normalize(tt.args.s); got != tt.want {
+				t.Errorf("normalize() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}