972 错误的思路

Author: aQuaYi

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

@@ -6,13 +6,15 @@ import (
 )
 
 func isRationalEqual(S string, T string) bool {
-	si, sn, sr := parse(S)
-	sn, sr = simplify(sn, sr)
-	s, sr := convert(si, sn, sr)
+	S = normalize(S)
+	sn, sr := parse2(S)
+	sn, sr = simplify2(sn, sr)
+	s, sr := convert2(sn, sr)
 
-	ti, tn, tr := parse(T)
-	tn, tr = simplify(tn, tr)
-	t, tr := convert(ti, tn, tr)
+	T = normalize(T)
+	tn, tr := parse2(T)
+	tn, tr = simplify2(tn, tr)
+	t, tr := convert2(tn, tr)
 
 	return s == t && sr == tr
 }
@@ -22,11 +24,31 @@ func normalize(s string) string {
 	if !strings.Contains(s, ".") {
 		s += "."
 	}
+	if strings.HasSuffix(s, ".") {
+		s += "0"
+	}
 	dot := strings.Index(s, ".")
 	if dot < 4 {
 		s = strings.Repeat("0", 4-dot) + s
 	}
-	return strings.Replace(s, ".", "", 1)
+	return strings.Replace("1"+s, ".", "", 1)
+}
+
+func parse2(s string) (string, string) {
+	if !strings.Contains(s, "(") {
+		return s, ""
+	}
+
+	i := strings.Index(s, "(")
+
+	nonRepeat := s[:i]
+	repeat := s[i+1 : len(s)-1]
+
+	if repeat == "0" {
+		repeat = ""
+	}
+
+	return nonRepeat, repeat
 }
 
 func parse(s string) (string, string, string) {
@@ -63,6 +85,34 @@ func parse(s string) (string, string, string) {
 	return integer, nonRepeat, repeat
 }
 
+func simplify2(nonRepeat, repeat string) (string, string) {
+	if repeat == "" {
+		return nonRepeat, repeat
+	}
+
+	if repeat == strings.Repeat(repeat[:1], len(repeat)) {
+		repeat = repeat[:1]
+	}
+
+	for repeat[:len(repeat)/2] == repeat[len(repeat)/2:] {
+		repeat = repeat[:len(repeat)/2]
+	}
+
+	for strings.HasSuffix(nonRepeat, repeat) {
+		nonRepeat = nonRepeat[:len(nonRepeat)-len(repeat)]
+	}
+
+	for i := 1; i < len(repeat); i++ {
+		if strings.HasSuffix(nonRepeat, repeat[i:]) {
+			repeat = repeat[i:] + repeat[:i]
+			nonRepeat = nonRepeat[:len(nonRepeat)-len(repeat)+i]
+			break
+		}
+	}
+
+	return nonRepeat, repeat
+}
+
 func simplify(nonRepeat, repeat string) (string, string) {
 	if repeat == "" {
 		return nonRepeat, repeat
@@ -91,6 +141,15 @@ func simplify(nonRepeat, repeat string) (string, string) {
 	return nonRepeat, repeat
 }
 
+func convert2(nonRepeat, repeat string) (int, string) {
+	i, _ := strconv.Atoi(nonRepeat)
+	if repeat == "9" {
+		i++
+		repeat = ""
+	}
+	return i, repeat
+}
+
 func convert(integer, nonRepeat, repeat string) (int, string) {
 	i, _ := strconv.Atoi(integer)
 	for j := len(nonRepeat); j > 0; j-- {

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

@@ -13,6 +13,12 @@ var tcs = []struct {
 	ans bool
 }{
 
+	{
+		"1.0",
+		"1",
+		true,
+	},
+
 	{
 		"350.(111)",
 		"350.(11)",
@@ -55,12 +61,6 @@ var tcs = []struct {
 		true,
 	},
 
-	{
-		"1.0",
-		"1",
-		true,
-	},
-
 	{
 		"0.(52)",
 		"0.5(25)",
@@ -167,19 +167,19 @@ func Test_normalize(t *testing.T) {
 		{
 			"1",
 			args{"1"},
-			"0001",
+			"10001",
 		},
 
 		{
 			"1234.",
 			args{"1234."},
-			"1234",
+			"11234",
 		},
 
 		{
 			"1234.(1)",
 			args{"1234.(1)"},
-			"1234(1)",
+			"11234(1)",
 		},
 
 		// Add test cases.