slightly shorten z function

1 files changed, 4 insertions, 5 deletions

diff --git a/string/z.cpp b/string/z.cpp
index f3824f3..c128e9d 100644
--- a/string/z.cpp
+++ b/string/z.cpp
@@ -1,11 +1,10 @@
 vector<int> Z(const string& s) {
     int n = sz(s);
-    vector<int> z(n, n);
-    int x = 0, y = 0;
-    for (int i = 1; i < n; i++) {
-        z[i] = max(0, min(z[i - x], y - i + 1));
+    vector<int> z(n);
+    for (int i = 1, x = 0; i < n; i++) {
+        z[i] = max(0, min(z[i - x], x + z[x] - i));
         while (i + z[i] < n && s[z[i]] == s[i + z[i]]) {
-            x = i, y = i + z[i], z[i]++;
+            x = i, z[i]++;
     }}
     return z;
 }