QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#278489 | #7876. Cyclic Substrings | daredsakura# | ML | 120ms | 353564kb | C++14 | 2.2kb | 2023-12-07 16:40:53 | 2023-12-07 16:40:54 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 6000000 + 5;
const int mod = 998244353;
int n;
struct pam {
int sz, tot, last;
int cnt[maxn], ch[maxn][10], len[maxn], fail[maxn];
char s[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 solve() {
long long ans = 0;
for (int i = sz; i >= 0; i--) {
cnt[fail[i]] += cnt[i];
}
}
} pam1, pam2; // namespace pam
char s[maxn];
void dfs(int u,int v){
if(v!=-1)pam1.cnt[u]-=pam2.cnt[v];
for(int i=0;i<=9;++i){
if(pam1.ch[u][i]!=0){
if(v!=-1&&pam2.ch[v][i]!=0){
dfs(pam1.ch[u][i],pam2.ch[v][i]);
}else{
dfs(pam1.ch[u][i],-1);
}
}
}
}
long long solve() {
long long ans=0;
for(int i=2;i<=pam1.sz;i++){
if(pam1.len[i]<=n){
ans=(ans+pam1.len[i]*pam1.cnt[i]%mod*pam1.cnt[i]%mod)%mod;
}
}
return ans;
}
signed main() {
cin >> n;
pam1.clear();
pam2.clear();
scanf("%s", s + 1);
for (int i = 1; i <= n; i++) {
pam1.insert(s[i]);
pam2.insert(s[i]);
}
for (int i = 1; i <= n; i++) {
pam1.insert(s[i]);
}
pam1.solve();
pam2.solve();
dfs(0,0);
dfs(1,1);
cout<<solve()<<endl;
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 7816kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 1ms
memory: 7568kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 1ms
memory: 9552kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 1ms
memory: 7572kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 1ms
memory: 5528kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 0ms
memory: 5468kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 1ms
memory: 7700kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: 0
Accepted
time: 120ms
memory: 353564kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
496166704
result:
ok 1 number(s): "496166704"
Test #9:
score: -100
Memory Limit Exceeded
input:
3000000 2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...
output:
890701718