小月赛 C 线性哈希做法!
upd:现在是严格线性了。
设
其中
把主串等效成
注意到
证明
而
诶那你求
因为
考虑用等效后的序列进行匹配,则
即
注意
且其他点中连续段仅有不超过 100 种不同的长度。
所以把
此时存在整数
顺便把
考虑预处理
其中每个序列长度为
匹配的复杂度显然线性。
#include <cstdio>
#include <unordered_map>
using namespace std;
int n, m, u, v, O, a[5000050], b[5000050], s[5000050], t[5000050], T[5000050];
unsigned long long B, p[5000050], h1[5000050], h2[5000050];
unordered_map<unsigned long long, int> V;
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1, x, l = -1, c = 0; i <= n; ++i)
{
scanf("%d", &x);
if (x == l)
++c;
else
{
if (~l)
a[++u] = l, s[u] = c;
l = x, c = 1;
}
if (i == n)
a[++u] = l, s[u] = c;
}
for (int i = 1, x, l = -1, c = 0; i <= m; ++i)
{
scanf("%d", &x);
if (x == l)
++c;
else
{
if (~l)
b[++v] = l, t[v] = c;
l = x, c = 1;
}
if (i == m)
b[++v] = l, t[v] = c;
}
if (v == 1)
{
long long q = 0;
for (int i = 1; i <= u; ++i)
if (a[i] == b[1])
q += s[i] * (s[i] + 1ll) >> 1;
return !printf("%lld", q);
}
int o = 1e9, g = 0;
for (int i = 1; i <= v; ++i)
o = min(o, t[i]);
for (g = o; g; --g)
{
bool f = 1;
for (int i = 1; i <= v; ++i)
if (t[i] % g)
{
f = 0;
break;
}
if (f)
break;
}
for (int i = 1; i <= v; ++i)
t[i] /= g;
if (v == 2)
{
long long q = 0;
for (int i = 1; i < u; ++i)
if (a[i] == b[1] && a[i + 1] == b[2])
q += min(s[i] / t[1], s[i + 1] / t[2]);
return !printf("%lld", q);
}
long long q = 0;
for (int i = 1; i <= u; ++i)
h1[i] = h1[i - 1] * 233 + a[i], h2[i] = h2[i - 1] * 233 + (T[s[i]] ? T[s[i]] : T[s[i]] = ++O);
for (int i = p[0] = 1; i <= v; ++i)
p[i] = p[i - 1] * 233, B = B * 233 + b[i], T[t[i]] ? T[t[i]] : T[t[i]] = ++O;
for (int i = 1; m * i <= n; ++i)
{
unsigned long long H = 0;
for (int j = 2; j < v; ++j)
H = H * 233 + T[t[j] * i];
V[H] = i;
}
if (O <= 100)
{
for (int i = 1, o; i <= u - v + 1; ++i)
if (h1[i + v - 1] - h1[i - 1] * p[v] == B && V.count(h2[i + v - 2] - h2[i] * p[v - 2]))
{
o = V[h2[i + v - 2] - h2[i] * p[v - 2]];
q += s[i] >= o * t[1] && s[i + v - 1] >= o * t[v];
}
return !printf("%lld", q);
}
else
{
for (int i = 1, f, o; i <= u - v + 1; ++i)
{
f = 1;
o = 0;
if (h1[i + v - 1] - h1[i - 1] * p[v] == B)
{
for (int j = i + 1; j < i + v - 1; ++j)
{
if (s[j] % t[j - i + 1] || o && o != s[j] / t[j - i + 1])
{
f = 0;
break;
}
if (!o)
o = s[j] / t[j - i + 1];
}
if (f && s[i] >= o * t[1] && s[i + v - 1] >= o * t[v])
++q;
}
}
return !printf("%lld", q);
}
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具