QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#672770 | #7876. Cyclic Substrings | gogo11 | AC ✓ | 731ms | 1034344kb | C++20 | 3.4kb | 2024-10-24 18:52:03 | 2024-10-24 18:52:04 |
Judging History
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,我给组数据试试?
Details
Tip: Click on the bar to expand more detailed information
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