summaryrefslogtreecommitdiff
path: root/content/string/z.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'content/string/z.cpp')
-rw-r--r--content/string/z.cpp10
1 files changed, 10 insertions, 0 deletions
diff --git a/content/string/z.cpp b/content/string/z.cpp
new file mode 100644
index 0000000..069fa38
--- /dev/null
+++ b/content/string/z.cpp
@@ -0,0 +1,10 @@
+vector<int> Z(const string& s) {
+ int n = sz(s);
+ 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, z[i]++;
+ }}
+ return z;
+}