Delicious Dessert
考虑先建出后缀自动机。
后缀自动机一个常用性质:每个状态表示所有字符串的 apbccdd
,最短是 cdd
,那么 ccdd
,bccdd
和 pbccdd
都属于这个状态中。
于是我们构造自动机后,从
#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;
}
分类:
题解
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现