[Luogu P4173] [BZOJ 4259] 残缺的字符串

最新推荐文章于 2022-01-08 21:16:34 发布

LPA20020220

最新推荐文章于 2022-01-08 21:16:34 发布

阅读量251

点赞数

分类专栏： FFT 字符串文章标签： FFT 字符串

本文链接：https://blog.csdn.net/LPA20020220/article/details/82896104

版权

字符串同时被 2 个专栏收录

36 篇文章 1 订阅

订阅专栏

FFT

7 篇文章 0 订阅

订阅专栏

题目描述

很久很久以前，在你刚刚学习字符串匹配的时候，有两个仅包含小写字母的字符串 $A$ 和 $B$ ，其中 $A$ 串长度为 $m$ ， $B$ 串长度为 $n$ 。可当你现在再次碰到这两个串时，这两个串已经老化了，每个串都有不同程度的残缺。

你想对这两个串重新进行匹配，其中 $A$ 为模板串，那么现在问题来了，请回答，对于 $B$ 的每一个位置 $i$ ，从这个位置开始连续 $m$ 个字符形成的子串是否可能与 $A$ 串完全匹配?

输入输出格式

输入格式：

第一行包含两个正整数 $m, n$ ，分别表示 $A$ 串和 $B$ 串的长度。

第二行为一个长度为 $m$ 的字符串 $A$ 。

第三行为一个长度为 $n$ 的字符串 $B$ 。

两个串均仅由小写字母和 $*$ 号组成，其中 $*$ 号表示相应位置已经残缺。

输出格式：

第一行包含一个整数 $k$ ，表示 $B$ 串中可以完全匹配 $A$ 串的位置个数。

若 $k > 0$ ，则第二行输出 $k$ 个正整数，从小到大依次输出每个可以匹配的开头位置（下标从 $1$ 开始）。

输入输出样例

输入样例#1：

3 7
a*b
aebr*ob

输出样例#1：

2
1 5

说明

100％的数据满足 $\leq m \leq n \leq 300000$ 。

解题分析

这种通配符匹配的问题 $D P$ 也能做，但显然要跑一年才能跑出来… 因为字符串不全自动机也就 $G G$ 了…

所以我们考虑换一种思路，如果没有通配符匹配的串会满足
$\sum_{i=1}^{m}(A[i]-B[pos+i])^2=0$
那么如果有通配符的话实际上其的权值设为0就行，然后我们魔改一下这个式子就行了：
$\sum_{i=1}^{m}A[i]\times(A[i]-B[pos+i])^2\times B[i]=0$
但这玩意还是没法做，我们把 $A$ 翻转一下上面那个式子就变成了一个卷积的形式，做 $7$ 遍 $F F T$ 就好了。

代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <cctype>
#define R register
#define IN inline
#define W while
#define db double
#define MX 2400500
const db PI = std::acos(-1.0);
struct Complex {db im, re;} a[MX], b[MX], c[MX];
IN Complex operator * (const Complex &x, const Complex &y)
{return {x.im * y.re + x.re * y.im, x.re * y.re - x.im * y.im};}
IN Complex operator + (const Complex &x, const Complex &y)
{return {x.im + y.im, x.re + y.re};}
IN Complex operator - (const Complex &x, const Complex &y)
{return {x.im - y.im, x.re - y.re};}
IN Complex operator * (const Complex &x, const db &y)
{return {x.im * y, x.re * y};}
char tar[MX], mod[MX];
int rev[MX], ans[MX];
db v1[MX], v2[MX];
int tot, sum, lg, cnt, len1, len2;
IN void FFT(Complex *dat, const int &typ)
{
	for (R int i = 1; i < tot; ++i) if(rev[i] < i) std::swap(dat[rev[i]], dat[i]);
	R int cur, seg, bd, step, now; Complex deal, base, buf1, buf2;
	for (seg = 1; seg < tot; seg <<= 1)
	{
		step = seg << 1; base = {std::sin(PI / seg) * typ, std::cos(PI / seg)};
		for (now = 0; now < tot; now += step)
		{
			deal = {0, 1}; bd = now + seg;
			for (cur = now; cur < bd; ++cur, deal = deal * base)
			{
				buf1 = dat[cur], buf2 = dat[cur + seg] * deal;
				dat[cur] = buf1 + buf2, dat[cur + seg] = buf1 - buf2;
			}
		}
	}
}
int main(void)
{
	scanf("%d%d", &len1, &len2);
	scanf("%s%s", tar, mod);
	for (R int i = 0; i < len1; ++i) v1[len1 - i - 1] = (tar[i] == '*') ? 0 : tar[i] - 'a' + 1;
	for (R int i = 0; i < len2; ++i) v2[i] = (mod[i] == '*') ? 0 : mod[i] - 'a' + 1;
	sum = len1 + len2;
	for (tot = 1; tot <= sum; tot <<= 1, ++lg); tot <<= 1, ++lg;
	for (R int i = 1; i < tot; ++i) rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (lg - 1));
	for (R int i = 0; i < tot; ++i) a[i] = {0, v1[i] * v1[i] * v1[i]}, b[i] = {0, v2[i]};
	FFT(a, 1), FFT(b, 1);
	for (R int i = 0; i < tot; ++i) c[i] = a[i] * b[i];
	for (R int i = 0; i < tot; ++i) a[i] = {0, v1[i] * v1[i]}, b[i] = {0, v2[i] * v2[i]};
	FFT(a, 1), FFT(b, 1);
	for (R int i = 0; i < tot; ++i) c[i] = c[i] - a[i] * b[i] * 2.0;
	for (R int i = 0; i < tot; ++i) a[i] = {0, v1[i]}, b[i] = {0, v2[i] * v2[i] * v2[i]};
	FFT(a, 1), FFT(b, 1);
	for (R int i = 0; i < tot; ++i) c[i] = c[i] + a[i] * b[i];
	FFT(c, -1); int bd = len2 - len1;
	for (R int i = 0; i <= bd; ++i) if(fabs(c[i + len1 - 1].re) < 0.5) ans[++cnt] = i + 1;
	printf("%d\n", cnt);
	for (R int i = 1; i <= cnt; ++i) printf("%d ", ans[i]);
}