#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, modulo = 1e9 + 7;
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) % modulo + (grid[curx][cury] - 'a' + 1), hsh1 %= modulo;
hsh2 = (hsh2*base2) % modulo + (grid[curx][cury] - 'a' + 1), hsh2 %= modulo;
if (k >= r) continue;
rhsh1 = (rhsh1*base1) % modulo + (grid[curx][cury] - 'a' + 1), rhsh1 %= modulo;
rhsh2 = (rhsh2*base1) % modulo + (grid[curx][cury] - 'a' + 1), rhsh2 %= modulo;
}
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;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 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 |
528 KB |
Output is correct |
6 |
Incorrect |
141 ms |
1980 KB |
Output isn't correct |
7 |
Incorrect |
2946 ms |
10796 KB |
Output isn't correct |
8 |
Execution timed out |
4083 ms |
9992 KB |
Time limit exceeded |