QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#274541 | #7876. Cyclic Substrings | Murphy_ | WA | 32ms | 247136kb | C++14 | 1.1kb | 2023-12-03 17:05:50 | 2023-12-03 17:05:52 |
Judging History
answer
#include<bits/stdc++.h>
#define ll long long
#define re register
#define il inline
#define N 6000006
#define mod 998244353
using namespace std;
char s[N];
int las,ch[N][26],len[N],fail[N],ct;
int cnt[N],tmp[N],n;
void init() {
ct=1;las=0;
s[0]='*';
len[0]=0;fail[0]=1;
len[1]=-1;fail[1]=1;
}
int getfail(int pos,int x) {while(s[pos-len[x]-1]!=s[pos]) x=fail[x];return x;}
void insert(int pos,char c) {
int nw=getfail(pos,las);
if(!ch[nw][c-'a']) {
len[++ct]=len[nw]+2;
fail[ct]=ch[getfail(pos,fail[nw])][c-'a'];
ch[nw][c-'a']=ct;
}
++cnt[las=ch[nw][c-'a']];
}
ll solve() {
ll ans=0;
for(int i=ct;i>=0;--i) tmp[fail[i]]+=tmp[i];
for(int i=ct;i>=0;--i) cnt[fail[i]]+=cnt[i];
for(int i=0;i<=ct;++i) cnt[i]-=tmp[i];
for(int i=2;i<=ct;++i) if(len[i]<=n) ans=(ans+1ll*(cnt[i]*(cnt[i]*len[i])%mod)%mod)%mod;
return ans;
}
int main()
{
cin>>n>>(s+1);
init();
for(int i=1;i<=n;++i) s[n+i]=s[i];
for(int i=1;i<=n;++i) insert(i,s[i]);
for(int i=0;i<=ct;++i) tmp[i]=cnt[i];
for(int i=n+1;i<=2*n;++i) insert(i,s[i]);
cout<<solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 11800kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 0ms
memory: 11572kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 0ms
memory: 11628kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 0ms
memory: 11636kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 1ms
memory: 11636kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 2ms
memory: 11540kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 0ms
memory: 11640kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: -100
Wrong Answer
time: 32ms
memory: 247136kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
-239992835
result:
wrong answer 1st numbers differ - expected: '496166704', found: '-239992835'