QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#639030 | #7876. Cyclic Substrings | kinoko777 | RE | 1ms | 9824kb | C++23 | 3.0kb | 2024-10-13 17:42:48 | 2024-10-13 17:42:48 |
Judging History
answer
#include <algorithm>
#include <cstring>
#include <iostream>
#include <string>
using namespace std;
constexpr int MAXN = 300000 + 5, mod = 998244353;
int n;
namespace pam {
int sz, tot, last, sz1;
int cnt[MAXN], ch[MAXN][10], len[MAXN], fail[MAXN];
char s[MAXN];
int cnt1[MAXN];
int node(int l) { // 建立一个新节点,长度为 l
sz++;
memset(ch[sz], 0, sizeof(ch[sz]));
len[sz] = l;
fail[sz] = cnt[sz] = 0;
return sz;
}
void clear() { // 初始化
sz = -1;
last = 0;
s[tot = 0] = '$';
node(0);
node(-1);
fail[0] = 1;
}
int getfail(int x) { // 找后缀回文
while (s[tot - len[x] - 1] != s[tot]) x = fail[x];
return x;
}
void insert(char c) { // 建树
s[++tot] = c;
int now = getfail(last);
if (!ch[now][c - '0']) {
int x = node(len[now] + 2);
fail[x] = ch[getfail(fail[now])][c - '0'];
ch[now][c - '0'] = x;
}
last = ch[now][c - '0'];
cnt[last]++;
}
void work1()
{
for(int i = 2; i <= sz; i ++) cnt1[i] = cnt[i];
for(int i = sz; i >= 2; i --) cnt1[fail[i]] += cnt1[i];
sz1 = sz;
}
int work(int op)
{
// int ans = 0;
// for(int i = sz; i >= 0; i --)
// {
// cnt[fail[i]] += cnt[i];
// }
int ans = 0;
for(int i = sz; i >= 2; i --)
{
cnt[fail[i]] += cnt[i];
int c = cnt[i];
if(i <= sz1) c -= cnt1[i];
if(len[i] <= n)
ans = (ans + c * c % mod * len[i] % mod) % mod;
}
ans = (ans % mod + mod) % mod;
// for(int i = 1; i <= sz; i ++)
// {
// if(len[i] == -1) continue;
// if(op == 2 && len[i] >= tot / 2) continue;
// cout << cnt[i] << ' ' << len[i] << endl;
// ans += 1ll * cnt[i] * cnt[i] % mod * len[i] % mod;
// ans %= mod;
// }
return ans;
}
}
string s;
int main() {
cin.tie(nullptr)->sync_with_stdio(false);
pam::clear();
cin >> n;
cin >> s;
pam::clear();
for(int i = 0; i < s.size(); i ++)
pam::insert(s[i]);
pam::work1();
for(int i = 0; i < s.size(); i ++)
pam::insert(s[i]);
cout << pam::work(1) << endl;
// vector<pair<int, int>> ccnt;
// for(int i = 2; i <= pam::sz; i ++)
// ccnt.push_back({pam :: cnt[i], pam :: len[i]});
// for(auto [c, l] : ccnt)
// cout << c << ' ' << l << endl;
// for(int i = 0; i < s.size(); i ++)
// pam::insert(s[i]);
// int now2 = pam::work(2);
// vector<pair<int, int>> ccnt2;
// for(int i = 2; i <= pam::sz; i ++)
// {
// if(pam::len[i] <= pam::tot / 2)
// ccnt2.push_back({pam :: cnt[i], pam :: len[i]});
// }
// for(auto [c, l] : ccnt2)
// {
// }
// cout << now2 << endl;
// cout << now2 - now1 << endl;
// pam1::clear();
// for(int i = 0; i < t.size(); i ++)
// pam1::insert(t[i]);
// int now2 = pam1::work(2);
// int now = ((now2 - now1) % mod + mod) % mod;
// cout << now << endl;
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 9816kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 1ms
memory: 7760kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 1ms
memory: 7764kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 1ms
memory: 7820kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 1ms
memory: 7768kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 1ms
memory: 7756kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 0ms
memory: 9824kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: -100
Runtime Error
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...