Advertisement
double_trouble

Z-function

Apr 22nd, 2016
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.65 KB | None | 0 0
  1. #include <iostream>
  2. #include <iomanip>
  3. #include <vector>
  4. #include <string>
  5. #include <cstring>
  6. #include <cmath>
  7. #include <algorithm>
  8. #include <set>
  9. #include <queue>
  10.  
  11. using namespace std;
  12.  
  13. const int inf = 2000000007;
  14.  
  15. int z[10000];
  16.  
  17. int main()
  18. {
  19.  
  20.     string s;
  21.     cin >> s;
  22.     int n = s.size();
  23.  
  24.     z[0] = 0;
  25.  
  26.     int l = 0, r = 0;
  27.     for (int i = 1; i < n && l < n && r < n; i++)
  28.     {
  29.         if (l <= i && i <= r)
  30.             z[i] = min(r - i + 1, z[i - l]);
  31.  
  32.         while (i + z[i] < n && s[i + z[i]] == s[z[i]])
  33.             z[i]++;
  34.  
  35.         if (i + z[i] - 1 > r)
  36.         {
  37.             r = i + z[i] - 1;
  38.             l = i;
  39.         }
  40.     }
  41.  
  42.     for (int i = 0; i < n; i++)
  43.         cout << z[i] << " ";
  44.  
  45.     return 0;
  46. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement