1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
| #include <bits/stdc++.h> using namespace std; #define int long long typedef unsigned long long ull;
const int N = 200010, P = 131;
int n, m, q; string str1, str2; map<int, int> cnt; ull h1[N], h2[N], p[N];
ull get(ull h[], int l, int r) { return h[r] - h[l - 1] * p[r - l + 1]; }
bool check(int i, int len) { if (i + len - 1 > n) return false; return get(h1, i, i + len - 1) == get(h2, 1, len); }
signed main() { cin >> n >> m >> q; cin >> str1 >> str2; str1 = " " + str1; str2 = " " + str2; p[0] = 1; int len = max(n, m); for (int i = 1; i <= len; i++) { p[i] = p[i - 1] * P; } for (int i = 1; i <= n; i++) { h1[i] = h1[i - 1] * P + str1[i]; } for (int i = 1; i <= m; i++) { h2[i] = h2[i - 1] * P + str2[i]; } for (int i = 1; i <= n; i++) { int l = 0; int r = m; int ans = 0; while (l <= r) { int mid = (l + r) / 2; if (check(i, mid)) { ans = mid; l = mid + 1; } else { r = mid - 1; } } cnt[ans]++; } while (q--) { int x; cin >> x; cout << cnt[x] << "\n"; } }
|