2023年ICPC亚洲合肥赛区赛 C. Cyclic Substrings

题目

题解

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
// #define int long long
#define ll long long
const int maxn = 6e6 + 5;
const int mod = 998244353;
int fail[maxn];//fail[i]表示i结点代表的回文串的最大回文后缀的编号 
int len[maxn]; //len[i]表示结点i代表的回文串的长度 
int trie[maxn][26], tot = 1;//tot初始为1!!!! 
int cnt[maxn];//结点i代表的回文串出现了多少次
int ind[maxn];
string s;
int get(int x, int i){//x是以s[i-1]结尾的回文串,返回以s[i-1]结尾且s[i-len[x]-1]==s[i]的回文串的对应结点编号 
	while(i - len[x] - 1 < 0 || s[i - len[x] - 1] != s[i]) x = fail[x];
	return x;
}
signed main(){
	int i, j;
	int n;
	cin >> n;
	cin >> s;
	s = s + s;
	int cur = 0;
	int pos, last = 0;
	fail[0] = 1;
	fail[1] = 0;
	len[0] = 0;
	len[1] = -1;
	for(i = 0; i < 2 * n; i++){
		pos = get(cur, i);//cur是以s[i-1]结尾的最大回文串 
		int idx = s[i] - '0';
		if(!trie[pos][idx]){ 
			tot++; 
			int t = get(fail[pos], i);
			fail[tot] = trie[t][idx];
			ind[trie[t][idx]]++;
			trie[pos][idx] = tot;//把新建的结点接在pos下面 
			len[tot] = len[pos] + 2;
			// cnt[tot] = cnt[fail[tot]] + 1;
		}
		cur = trie[pos][idx];
		if(i > n - 1)//只计算以 后半串的字符 结尾的最长回文串,否则会重复
			cnt[cur]++;//结点cur代表的回文串出现了多少次
	}
	
	ll res = 0;
	queue<int> q;
	for(int i = 2; i <= tot; i++){
		// if(len[i] > n) continue;
		if(!ind[i]){
			q.push(i);
		}
	}
	while(!q.empty()){
		int u = q.front();
		q.pop();
		int v = fail[u];
		// if(len[v] > n) continue;
		cnt[v] += cnt[u];//一个回文串出现一次,那么它的回文后缀也出现一次
		if(--ind[v] == 0){
			q.push(v);
		}
	}
	for(int i = 2; i <= tot; i++){
		if(len[i] > n) continue;
		// cout << i << ' ' << len[i] << ' ' << cnt[i] << '\n';
		res = (res + 1LL * len[i] * cnt[i] % mod * cnt[i] % mod) % mod;
	}
	cout << res << '\n';
	return 0;
}
相关推荐
代码AC不AC5 分钟前
【数据结构】队列
c语言·数据结构·学习·队列·深度讲解
林泽毅5 分钟前
SwanLab x EasyR1:多模态LLM强化学习后训练组合拳,让模型进化更高效
算法·llm·强化学习
小林熬夜学编程7 分钟前
【高并发内存池】第八弹---脱离new的定长内存池与多线程malloc测试
c语言·开发语言·数据结构·c++·算法·哈希算法
刚入门的大一新生14 分钟前
归并排序延伸-非递归版本
算法·排序算法
独好紫罗兰18 分钟前
洛谷题单3-P1980 [NOIP 2013 普及组] 计数问题-python-流程图重构
开发语言·python·算法
独好紫罗兰23 分钟前
洛谷题单3-P1009 [NOIP 1998 普及组] 阶乘之和-python-流程图重构
开发语言·python·算法
曦月逸霜35 分钟前
蓝桥杯高频考点——高精度(含C++源码)
c++·算法·蓝桥杯
ゞ 正在缓冲99%…44 分钟前
leetcode152.乘积最大子数组
数据结构·算法·leetcode
闯闯爱编程1 小时前
数组与特殊压缩矩阵
数据结构·算法·矩阵
秋风战士2 小时前
通信算法之255:无人机频谱探测设备技术详解
算法·无人机