QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#672770#7876. Cyclic Substringsgogo11AC ✓731ms1034344kbC++203.4kb2024-10-24 18:52:032024-10-24 18:52:04

Judging History

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

  • [2024-10-24 18:52:04]
  • 评测
  • 测评结果:AC
  • 用时:731ms
  • 内存:1034344kb
  • [2024-10-24 18:52:03]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
constexpr int Mod=998244353;

struct PAM {
    static constexpr int ALPHABET_SIZE=10;
    struct Node {
        int len,link,cnt;
        array<int,ALPHABET_SIZE> next;
        Node():len{},link{},cnt{},next{} {}
    };
    vector<Node> t;
    int suff;
    string s;
    PAM() {init();}
    void init() {
        t.assign(2,Node());
        t.reserve(4000000);
        t[0].len=-1;
        suff=1;
        s.clear();
    }
    int newNode() {
        t.emplace_back();
        return t.size()-1;
    }
    bool add(char c) {
        int pos=s.size();
        s+=c;
        int let=c-'0';
        int cur=suff,curlen=0;
        while(true) {
            curlen=t[cur].len;
            if(pos-1-curlen>=0 && s[pos-1-curlen]==s[pos]) {
                break;
            }
            cur=t[cur].link;
        }
        if(t[cur].next[let]) {
            suff=t[cur].next[let];
            return false;
        }
        int num=newNode();
        suff=num;
        t[num].len=t[cur].len+2;
        t[cur].next[let]=num;
        if(t[num].len==1) {
            t[num].link=1;
            t[num].cnt=1;
            return true;
        }
        while(true) {
            cur=t[cur].link;
            curlen=t[cur].len;
            if(pos-1-curlen>=0 && s[pos-1-curlen]==s[pos]) {
                t[num].link=t[cur].next[let];
                break;
            }
        }
        t[num].cnt=1+t[t[num].link].cnt;
        return true;
    }
    int next(int p,int x) {return t[p].next[x];};
    int link(int p) {return t[p].link;}
    int len(int p) {return t[p].len;}
    int cnt(int p) {return t[p].cnt;}
    int size() {return t.size();}
}pam;
void Add(int &a,int b) {
    a+=b;
    if(a>=Mod) a-=Mod;
}
int add(int a,int b) {
    Add(a,b);
    return a;
}
void Mul(int &a,int b) {
    a=1LL*a*b%Mod;
}
int mul(int a,int b) {
    Mul(a,b);
    return a;
}
void solve() {
    int n;
    cin>>n;
    string s;
    cin>>s;
    vector<int> ends;
    for(auto ch: s) {
        pam.add(ch);
        ends.push_back(pam.suff);
    }
    int x1=pam.size();
    vector<int> f1(x1);
    for(int v: ends)
        f1[v]++;
    vector adj(x1,vector<int>());
    for(int i=2;i<x1;i++) {
        adj[pam.link(i)].push_back(i);
    }
    auto dfs1=[&](auto &&dfs1,int u)->void {
        for(int v: adj[u]) {
            dfs1(dfs1,v);
            f1[u]+=f1[v];
        }
        // cout<<pam.len(u)<<"    "<<f[u]<<"\n";
        // if(pam.len(u)<=n) Add(ans,mul(pam.len(u),mul(f[u],f[u])));
    };
    dfs1(dfs1,1);
    for(auto ch: s) {
        pam.add(ch);
        ends.push_back(pam.suff);
    }
    int x2=pam.size();
    adj.resize(x2);
    vector<int> f2(x2);
    for(int v: ends)
        f2[v]++;
    for(int i=x1;i<x2;i++) {
        adj[pam.link(i)].push_back(i);
    }
    int ans=0;
    auto dfs2=[&](auto &&dfs2,int u)->void {
        for(int v: adj[u]) {
            dfs2(dfs2,v);
            f2[u]+=f2[v];
        }
        // cout<<pam.len(u)<<"    "<<f[u]<<"\n";
        if(pam.len(u)<=n && u<x1) Add(ans,mul(pam.len(u),mul(f2[u]-f1[u],f2[u]-f1[u])));
        else if(pam.len(u)<=n) Add(ans,mul(pam.len(u),mul(f2[u],f2[u])));
    };
    dfs2(dfs2,1);
    cout<<ans<<"\n";
}
int main() {
    ios::sync_with_stdio(false),cin.tie(0);
    int _=1;
    // cin>>_;
    while(_--)
        solve();
    return 0;
}

这程序好像有点Bug,我给组数据试试?

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 3488kb

input:

5
01010

output:

39

result:

ok 1 number(s): "39"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3852kb

input:

8
66776677

output:

192

result:

ok 1 number(s): "192"

Test #3:

score: 0
Accepted
time: 0ms
memory: 3536kb

input:

1
1

output:

1

result:

ok 1 number(s): "1"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3624kb

input:

2
22

output:

12

result:

ok 1 number(s): "12"

Test #5:

score: 0
Accepted
time: 0ms
memory: 3608kb

input:

2
21

output:

2

result:

ok 1 number(s): "2"

Test #6:

score: 0
Accepted
time: 0ms
memory: 3604kb

input:

3
233

output:

10

result:

ok 1 number(s): "10"

Test #7:

score: 0
Accepted
time: 0ms
memory: 3792kb

input:

3
666

output:

54

result:

ok 1 number(s): "54"

Test #8:

score: 0
Accepted
time: 134ms
memory: 349380kb

input:

1000000
3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...

output:

496166704

result:

ok 1 number(s): "496166704"

Test #9:

score: 0
Accepted
time: 374ms
memory: 1034344kb

input:

3000000
2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...

output:

890701718

result:

ok 1 number(s): "890701718"

Test #10:

score: 0
Accepted
time: 220ms
memory: 365880kb

input:

3000000
9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999...

output:

224009870

result:

ok 1 number(s): "224009870"

Test #11:

score: 0
Accepted
time: 426ms
memory: 869636kb

input:

3000000
8989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989898989...

output:

51985943

result:

ok 1 number(s): "51985943"

Test #12:

score: 0
Accepted
time: 525ms
memory: 816128kb

input:

3000000
1911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911911...

output:

355676465

result:

ok 1 number(s): "355676465"

Test #13:

score: 0
Accepted
time: 345ms
memory: 613384kb

input:

3000000
7777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777777...

output:

788510374

result:

ok 1 number(s): "788510374"

Test #14:

score: 0
Accepted
time: 342ms
memory: 595212kb

input:

3000000
5555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555...

output:

691884476

result:

ok 1 number(s): "691884476"

Test #15:

score: 0
Accepted
time: 711ms
memory: 396780kb

input:

3000000
0990990909909909099090990990909909909099090990990909909099099090990990909909099099090990990909909099099090990909909909099099090990909909909099090990990909909909099090990990909909909099090990990909909099099090990990909909099099090990990909909099099090990909909909099099090990909909909099090990...

output:

701050848

result:

ok 1 number(s): "701050848"

Test #16:

score: 0
Accepted
time: 236ms
memory: 148648kb

input:

3000000
2772772727727727277272772772727727727277272772772727727277277272772772727727277277272772772727727277277272772727727727277277272772727727727277272772772727727727277272772772727727727277272772772727727277277272772772727727277277272772772727727277277272772727727727277277272772727727727277272772...

output:

486861605

result:

ok 1 number(s): "486861605"

Test #17:

score: 0
Accepted
time: 731ms
memory: 521792kb

input:

3000000
4554554545545545455454554554545545545455454554554545545455455454554554545545455455454554554545545455455454554545545545455455454554545545545455454554554545545545455454554554545545545455454554554545545455455454554554545545455455454554554545545455455454554545545545455455454554545545545455454554...

output:

450625621

result:

ok 1 number(s): "450625621"

Test #18:

score: 0
Accepted
time: 666ms
memory: 515140kb

input:

3000000
1181811811818118181181181811818118118181181181811818118118181181181811818118118181181811811818118118181181811811818118181181181811811818118181181181811811818118181181181811818118118181181181811818118118181181181811818118118181181811811818118118181181811811818118181181181811811818118181181181...

output:

649551870

result:

ok 1 number(s): "649551870"

Extra Test:

score: 0
Extra Test Passed