#include <bits/stdc++.h>
using namespace std;
#define int long long
int gcd(int a, int b) {
return !b ? a : gcd(b, a % b);
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m, l;
cin >> n >> m >> l;
vector<vector<char>> grid(n + 5, vector<char>(m + 5));
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
cin >> grid[i][j];
}
}
vector<int> dx = {-1, -1, 0, 1, 1, 1, 0, -1}, dy = {0, 1, 1, 1, 0, -1, -1, -1};
map<tuple<int, int, int, int>, int> cnt;
int sz = min(l, max(n, m)), r = l % sz, total = 0;
const int base1 = 31, base2 = 53, modulo1 = 1e9 + 7, modulo2 = 1e9 + 9;
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
for (int d = 0; d < 8; d++) {
int hsh1 = 0, rhsh1 = 0, hsh2 = 0, rhsh2 = 0;
for (int k = 0; k < sz; k++) {
int curx = (i + (k*dx[d]) % n + n) % n, cury = (j + (k*dy[d]) % m + m) % m;
hsh1 = (hsh1*base1) % modulo1 + (grid[curx][cury] - 'a' + 1), hsh1 %= modulo1;
hsh2 = (hsh2*base2) % modulo2 + (grid[curx][cury] - 'a' + 1), hsh2 %= modulo2;
if (k >= r) continue;
rhsh1 = (rhsh1*base1) % modulo1 + (grid[curx][cury] - 'a' + 1), rhsh1 %= modulo1;
rhsh2 = (rhsh2*base2) % modulo2 + (grid[curx][cury] - 'a' + 1), rhsh2 %= modulo2;
}
cnt[tuple<int, int, int, int>(hsh1, rhsh1, hsh2, rhsh2)]++;
total++;
}
}
}
int ans = 0;
for (auto&x : cnt) ans += x.second*x.second;
total *= total;
int d = gcd(ans, total);
ans /= d, total /= d;
cout << ans << "/" << total << '\n';
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
2 ms |
340 KB |
Output is correct |
5 |
Correct |
12 ms |
468 KB |
Output is correct |
6 |
Incorrect |
138 ms |
2004 KB |
Output isn't correct |
7 |
Incorrect |
2915 ms |
10736 KB |
Output isn't correct |
8 |
Execution timed out |
4054 ms |
9604 KB |
Time limit exceeded |