Delicious Dessert

考虑先建出后缀自动机。

后缀自动机一个常用性质:每个状态表示所有字符串的 Endpos\operatorname{Endpos} 集合完全相等,也就是出现次数完全相等。另外一个结论,每个状态表示的所有字符串,一定是其字符串中最短和最长字符串之间。即对于构建自动机的 lenilen_i,即每个状态中所有字符串最长的长度,一定有 ii 对应的字符串是 [lenfai+1,leni][len_{fa_i}+1, len_i],即如果最长是 apbccdd,最短是 cdd,那么 ccddbccddpbccdd 都属于这个状态中。

于是我们构造自动机后,从 faifa_iii 连边之后 DFS,可以求出每个状态的 Endpos\operatorname{Endpos} 集合大小,即共同的出现次数,设为 fif_i。接着只需要求 [lenfai+1,leni][len_{fa_i}+1, len_i] 中有多少个数是 fif_i 的因数。容易发现可以预处理因数然后二分。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
using namespace std;

const int N = 2e6 + 5;

string s;
long long f[N];
int tot = 1, last = 1;
vector<int> G[N], kk[N];

struct Node
{
	int fa, len, son[26];
}g[N];

void extend(int c)
{
	int p = last;
	int np = last = ++tot;
	g[np].len = g[p].len + 1;
	f[np] = 1LL;
	for (; p && g[p].son[c] == 0; p = g[p].fa) g[p].son[c] = np;
	if (!p) g[np].fa = 1;
	else
	{
		int q = g[p].son[c];
		if (g[q].len == g[p].len + 1) g[np].fa = q;
		else
		{
			int nq = ++tot;
			g[nq] = g[q];
			g[nq].len = g[p].len + 1;
			g[q].fa = g[np].fa = nq;
			for (; p && g[p].son[c] == q; p = g[p].fa) g[p].son[c] = nq;
		}
	}
}

void dfs(int u)
{
	for (int j : G[u])
	{
		dfs(j);
		f[u] += f[j];
	}
}

int main()
{
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	for (long long i = 1; i <= (long long)1e6; i++)
	{
		for (long long j = i; j <= (long long)1e6; j += i)
		{
			kk[j].emplace_back(i);
		}
	}
	long long ans = 0;
	cin >> s >> s;
	for (char i : s) extend(i - 'a');
	for (int i = 2; i <= tot; i++) G[g[i].fa].emplace_back(i);
	dfs(1);
	for (int i = 2; i <= tot; i++)
	{
		int l = g[g[i].fa].len + 1, r = g[i].len; // 长度
		long long k = f[i]; // 次数
		long long cnt = upper_bound(kk[k].begin(), kk[k].end(), r) - lower_bound(kk[k].begin(), kk[k].end(), l);
		ans += cnt * k;
	}
	cout << ans << "\n";
	return 0;
}
posted @   HappyBobb  阅读(2)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示