QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#274572 | #7876. Cyclic Substrings | CSUST_GXL | RE | 36ms | 241504kb | C++20 | 1.4kb | 2023-12-03 17:47:08 | 2023-12-03 17:47:09 |
Judging History
answer
#include<bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int maxn=3e6+5;
const int mod=998244353;
int ch[maxn];
int tree[maxn][10],cnt[maxn],len[maxn],fail[maxn],trans[maxn],cmt[maxn];
int p,last,n;
int new_node(int id){
for(int i=0;i<10;i++) tree[p][i]=0;
cnt[p]=0;
len[p]=id;
return p++;
}
int nn;
void init(){
p=n=0;
new_node(0);
new_node(-1);
last=0;
fail[0]=1;
ch[0]=-1;
}
int get_fail(int x){
while(ch[n-len[x]-1]!=ch[n])x=fail[x];
return x;
}
int res=0;
void add(int c){
ch[++n]=c;
int cur=get_fail(last);
if(!tree[cur][c]){
int now= new_node(len[cur]+2);
fail[now]=tree[get_fail(fail[cur])][c];
tree[cur][c]=now;
}
last=tree[cur][c];
cnt[last]++;
}
void solve(){
cin>>nn;
init();
string s;
cin>>s;
for(auto i:s)add(i-'0');
long long anss=0;
for(int i=p;i>=1;i--){
cmt[i]+=cnt[i];
cmt[fail[i]]+=cmt[i];
}
for(auto i:s)add(i-'0');
for(int i=p;i>=1;i--)cnt[fail[i]]+=cnt[i];
long long aans=0;
for(int i=1;i<=p;i++){
cnt[i]-=cmt[i];
// cout<<cmt[i]<<" "<<cnt[i]<<" "<<len[i]<<endl;
if(len[i]<=nn) aans=(aans+1ll*cnt[i]*cnt[i]%mod*len[i]%mod)%mod;
}
cout<<aans<<endl;
}
signed main() {
int t = 1;
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
while (t--) solve();
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 14080kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 0ms
memory: 13736kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 0ms
memory: 14084kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 0ms
memory: 13828kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 0ms
memory: 13892kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 0ms
memory: 13856kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 2ms
memory: 14076kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: 0
Accepted
time: 36ms
memory: 241504kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
496166704
result:
ok 1 number(s): "496166704"
Test #9:
score: -100
Runtime Error
input:
3000000 2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...