QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#589176 | #7876. Cyclic Substrings | limpid | ML | 115ms | 511268kb | C++14 | 1.5kb | 2024-09-25 16:32:36 | 2024-09-25 16:32:36 |
Judging History
answer
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define int long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define mod 998244353
using namespace std;
inline int read(){
int f=1,ans=0; char c=getchar();
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){ans=ans*10+c-'0';c=getchar();}
return f*ans;
}
const int MAXN = 6e6 + 11;
int ch[MAXN][10], cnt, tot, fail[MAXN], lst, len[MAXN], siz[MAXN];
int node(int l){cnt++; len[cnt] = l; return cnt;}
char str[MAXN];
vector<int> vec[MAXN];
void Init(){
cnt = -1; str[0] = '&';
node(0), node(-1); fail[0] = 1;
lst = 0;
}
int getfail(int x, char c){
while(str[tot - len[x] - 1] != c){
x = fail[x];
}
return x;
}
int N, ret;
void ins(char c, int id){
tot++; int now = getfail(lst, c);
if(!ch[now][c - '0']){
int x = node(len[now] + 2);
fail[x] = ch[getfail(fail[now], c)][c - '0'];
ch[now][c - '0'] = x;
}lst = ch[now][c - '0'];
if(id > N) siz[lst]++;
}
void dfs(int u){
for(auto v: vec[u]) dfs(v), siz[u] += siz[v];
if(len[u] <= N && len[u] >= 1) ret += siz[u] * siz[u] % mod * len[u] % mod, ret %= mod;
return;
}
signed main(){
N = read(); Init();
scanf("%s", str + 1); for(int i = N + 1; i <= 2 * N; i++) str[i] = str[i - N];
for(int i = 1; i <= 2 * N; i++) ins(str[i], i);
for(int i = 0; i <= cnt; i++) if(i != 1) vec[fail[i]].pb(i);
dfs(1);
printf("%lld\n", ret);
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 15ms
memory: 154568kb
input:
5 01010
output:
39
result:
ok 1 number(s): "39"
Test #2:
score: 0
Accepted
time: 16ms
memory: 153720kb
input:
8 66776677
output:
192
result:
ok 1 number(s): "192"
Test #3:
score: 0
Accepted
time: 24ms
memory: 152824kb
input:
1 1
output:
1
result:
ok 1 number(s): "1"
Test #4:
score: 0
Accepted
time: 12ms
memory: 154748kb
input:
2 22
output:
12
result:
ok 1 number(s): "12"
Test #5:
score: 0
Accepted
time: 15ms
memory: 153332kb
input:
2 21
output:
2
result:
ok 1 number(s): "2"
Test #6:
score: 0
Accepted
time: 16ms
memory: 154040kb
input:
3 233
output:
10
result:
ok 1 number(s): "10"
Test #7:
score: 0
Accepted
time: 12ms
memory: 154760kb
input:
3 666
output:
54
result:
ok 1 number(s): "54"
Test #8:
score: 0
Accepted
time: 115ms
memory: 511268kb
input:
1000000 3333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333333...
output:
496166704
result:
ok 1 number(s): "496166704"
Test #9:
score: -100
Memory Limit Exceeded
input:
3000000 2222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222222...
output:
890701718