QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#283007 | #6567. Repetitive String Invention | warner1129# | TL | 1ms | 3548kb | C++20 | 3.4kb | 2023-12-13 17:41:31 | 2023-12-13 17:41:32 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
template <class... T> void dbg(T... x) { char e{}; ((cerr << e << x, e = ' '), ...); }
template <class T> void org(T l, T r) { while (l != r) cerr << ' ' << *l++; cerr << '\n'; }
#define debug(x...) dbg(#x, '=', x, '\n')
#define olist(x...) dbg(#x, '='), org(x)
#else
#define debug(...) ((void)0)
#define olist(...) ((void)0)
#endif
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()
#define ff first
#define ss second
using u32 = unsigned int;
using i64 = long long;
using u64 = unsigned long long;
using i128 = __int128;
using u128 = unsigned __int128;
using Pt = pair<int, int>;
template <class T> inline constexpr T inf = numeric_limits<T>::max() / 2;
constexpr int mod = 998244353;
template<class T> bool chmin(T &a, T b) { return (b < a and (a = b, true)); }
template<class T> bool chmax(T &a, T b) { return (a < b and (a = b, true)); }
template<class... T> int add(T... x) { int t{}; return (((t += x) %= mod), ...), t; }
template<class... T> int mul(T... x) { i64 t{1}; return (((t *= x) %= mod), ...), t; }
constexpr u64 P = 1e9 + 321;
void solve() {
string S;
cin >> S;
const int n = S.size();
vector<u64> pw(n + 1, 1);
for (int i = 1; i <= n; i++) {
pw[i] = pw[i - 1] * P;
}
auto build = [&](string_view) -> vector<u64> {
vector<u64> h(n);
for (int i = 0; i < n; i++) {
h[i] = S[i] * pw[i];
if (i) h[i] += h[i - 1];
}
return h;
};
auto get = [&](const vector<u64> &h, int l, int r) -> u64 {
assert(l <= r);
return (h[r] - (l == 0 ? 0 : h[l - 1])) * pw[n - l];
};
auto cal = [&](string_view S, bool same) -> i64 {
auto h = build(S);
i64 ans = 0;
for (int i = n - 1; i >= 1; i--) { // [0, i - 1] [i, n - 1]
map<u64, i64> cnt;
for (int j = i; j < n; j++) {
cnt[get(h, i, j)]++;
}
for (int l = 0; l < i; l++) {
string tmp;
vector<int> fail;
int p = -1;
for (int r = l; r < i; r++) {
tmp += S[r];
while (p != -1 and tmp[p + 1] != tmp.back()) {
p = fail[p];
}
if (tmp[p + 1] == tmp.back()) p++;
if (r == l) p = -1;
fail.push_back(p);
debug(l, r, tmp);
olist(all(fail));
for (int g = p; g != -1; g = fail[g]) {
int len = g + 1;
if (l + g + 1 <= r - g - 1) {
debug(l + g + 1, r - g - 1, cnt[get(h, l + g + 1, r - g - 1)]);
ans += cnt[get(h, l + g + 1, r - g - 1)];
}
}
if (same) {
debug(same, cnt[get(h, l, r)]);
ans += cnt[get(h, l, r)];
}
}
}
}
return ans;
};
i64 ans = cal(S, 1);
reverse(all(S));
ans += cal(S, 0);
cout << ans << '\n';
}
signed main() {
cin.tie(0)->sync_with_stdio(false);
cin.exceptions(cin.failbit);
int T = 1;
// cin >> T;
while (T--) {
solve();
}
return 0;
}
详细
Test #1:
score: 100
Accepted
time: 1ms
memory: 3548kb
input:
aaaa
output:
9
result:
ok single line: '9'
Test #2:
score: 0
Accepted
time: 1ms
memory: 3392kb
input:
axabxbcxcdxd
output:
22
result:
ok single line: '22'
Test #3:
score: -100
Time Limit Exceeded
input:
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx...