Update distinct-echo-substrings.cpp

kamyu104 · web-flow · commit 8ff64643bc90 · 2020-01-13T08:59:27.000+08:00
diff --git a/C++/distinct-echo-substrings.cpp b/C++/distinct-echo-substrings.cpp
@@ -6,8 +6,11 @@ class Solution {
     int distinctEchoSubstrings(string text) {
         unordered_set<string> result;
         int l = text.length() - 1;
-        for (int i = 0; i < min(2 * l, int(text.length())); ++i)  {  // aaaaaaabcdefabcdefabcdef
-            l = min(l, KMP(text, i, &result));
+        for (int i = 0; i < l; ++i)  {  // aaaaaaaaaaaaaaaaaaaaaaaaaaaaaabcdefabcdefabcdef
+            const auto& substr_len = KMP(text, i, &result);
+            if (substr_len != numeric_limits<int>::max()) {
+                l = min(l, i + substr_len);
+            }
         }
         return result.size();
     }

Original file line number	Diff line number	Diff line change
`@@ -6,8 +6,11 @@ class Solution {`
`6`	`6`	`int distinctEchoSubstrings(string text) {`
`7`	`7`	`unordered_set<string> result;`
`8`	`8`	`int l = text.length() - 1;`
`9`		`- for (int i = 0; i < min(2 * l, int(text.length())); ++i) { // aaaaaaabcdefabcdefabcdef`
`10`		`- l = min(l, KMP(text, i, &result));`
	`9`	`+ for (int i = 0; i < l; ++i) { // aaaaaaaaaaaaaaaaaaaaaaaaaaaaaabcdefabcdefabcdef`
	`10`	`+ const auto& substr_len = KMP(text, i, &result);`
	`11`	`+ if (substr_len != numeric_limits<int>::max()) {`
	`12`	`+ l = min(l, i + substr_len);`
	`13`	`+ }`
`11`	`14`	`}`
`12`	`15`	`return result.size();`
`13`	`16`	`}`