QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#589176#7876. Cyclic SubstringslimpidML 115ms511268kbC++141.5kb2024-09-25 16:32:362024-09-25 16:32:36

Judging History

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

  • [2024-09-25 16:32:36]
  • 评测
  • 测评结果:ML
  • 用时:115ms
  • 内存:511268kb
  • [2024-09-25 16:32:36]
  • 提交

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

result: