QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#274572#7876. Cyclic SubstringsCSUST_GXLRE 36ms241504kbC++201.4kb2023-12-03 17:47:082023-12-03 17:47:09

Judging History

你现在查看的是最新测评结果

  • [2023-12-03 17:47:09]
  • 评测
  • 测评结果:RE
  • 用时:36ms
  • 内存:241504kb
  • [2023-12-03 17:47:08]
  • 提交

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...

output:


result: