diff options
| author | mzuenni <michi.zuendorf@gmail.com> | 2022-06-27 17:19:28 +0200 |
|---|---|---|
| committer | mzuenni <michi.zuendorf@gmail.com> | 2022-06-27 17:19:28 +0200 |
| commit | 5ab8a5088b729a9953b8dff1b2a985dc8fb2098b (patch) | |
| tree | ed40d6936c0e9eee40ba62751cbf99ecddbaddc2 /string/z.cpp | |
| parent | adabbad9c51cf7cd3874bfde8eac1fbcf84fec10 (diff) | |
updated tcr
Diffstat (limited to 'string/z.cpp')
| -rw-r--r-- | string/z.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/string/z.cpp b/string/z.cpp new file mode 100644 index 0000000..dbd5ce5 --- /dev/null +++ b/string/z.cpp @@ -0,0 +1,15 @@ +vector<int> Z(const vector<int> &s) { + int n = sz(s); + vector<int> z(n, n); + int l = 0, r = 0, p, q; + for (int i = 1; i < n; ++i) { + if (i <= r && z[i - l] < r - i + 1) { + z[i] = z[i - l]; + } else { + if (i > r) p = 0, q = i; + else p = r - i + 1, q = r + 1; + while (q < n && s[p] == s[q]) ++p, ++q; + z[i] = q - i, l = i, r = q - 1; + }} + return z; +} |
