【CodeForces528D】Fuzzy Search

最新推荐文章于 2021-10-13 21:54:39 发布
最新推荐文章于 2021-10-13 21:54:39 发布
阅读量350 收藏
点赞数
分类专栏： 【OJ】CodeForces 【类型】做题记录 【算法】FFT与NTT
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_39972971/article/details/79525431
版权

【题目链接】

【思路要点】

  • FFT匹配字符串的问题，由于字符集较小，我们可以分别检验每个字符是不是匹配。
  • 具体做法就是把T翻转，然后FFT，检验每个位置匹配的字符数是否等等于该字符出现次数。
  • 时间复杂度\(O(|S|Log|S|)\)。

【代码】

/*Advanced FFT with smaller constant*/
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 524288;
const int MAXC = 256;
const double eps = 1e-3;
const double pi = acos(-1);
template <typename T> void chkmax(T &x, T y) {x = max(x, y); }
template <typename T> void chkmin(T &x, T y) {x = min(x, y); } 
template <typename T> void read(T &x) {
	x = 0; int f = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
	for (; isdigit(c); c = getchar()) x = x * 10 + c - '0';
	x *= f;
}
template <typename T> void write(T x) {
	if (x < 0) x = -x, putchar('-');
	if (x > 9) write(x / 10);
	putchar(x % 10 + '0');
}
template <typename T> void writeln(T x) {
	write(x);
	puts("");
}
struct point {double x, y; };
point operator + (point a, point b) {return (point) {a.x + b.x, a.y + b.y}; }
point operator - (point a, point b) {return (point) {a.x - b.x, a.y - b.y}; }
point operator * (point a, point b) {return (point) {a.x * b.x - a.y * b.y, a.x * b.y + a.y * b.x}; }
point operator / (point a, double x) {return (point) {a.x / x, a.y / x}; }
int sa[MAXN], sc[MAXN], sg[MAXN], st[MAXN];
int ta[MAXN], tc[MAXN], tg[MAXN], tt[MAXN];
point res[MAXN]; char s[MAXN], t[MAXN];
int n, m, k, N, Log, cnt[MAXC], home[MAXN];
bool ans[MAXN];
void FFTinit() {
	for (int i = 0; i < N; i++) {
		int tmp = i, ans = 0;
		for (int j = 1; j <= Log; j++) {
			ans <<= 1;
			ans += tmp & 1;
			tmp >>= 1;
		}
		home[i] = ans;
	}
}
void FFT(point *a, int mode) {
	for (int i = 0; i < N; i++)
		if (home[i] < i) swap(a[i], a[home[i]]);
	for (int len = 2; len <= N; len <<= 1) {
		point delta = (point) {cos(2 * pi / len * mode), sin(2 * pi / len * mode)};
		for (int i = 0; i < N; i += len) {
			point now = (point) {1, 0};
			for (int j = i, k = i + len / 2; k < i + len; j++, k++) {
				point tmp = a[j];
				point tnp = a[k] * now;
				a[j] = tmp + tnp;
				a[k] = tmp - tnp;
				now = now * delta;
			}
		}
	}
	if (mode == -1) {
		for (int i = 0; i < N; i++)
			a[i] = a[i] / (4 * N);
	}
}
int main() {
	read(n), read(m), read(k);
	scanf("\n%s\n%s", s, t);
	N = 1, Log = 0;
	while (N <= n) {
		N <<= 1;
		Log++;
	}
	reverse(t, t + m);
	for (int i = 0; i < m; i++) {
		if (t[i] == 'A') ta[i] = 1;
		if (t[i] == 'C') tc[i] = 1;
		if (t[i] == 'G') tg[i] = 1;
		if (t[i] == 'T') tt[i] = 1;
	}
	for (int i = 0; i < n; i++) {
		cnt[s[i]]++;
		if (cnt['A']) sa[i] = 1;
		if (cnt['C']) sc[i] = 1;
		if (cnt['G']) sg[i] = 1;
		if (cnt['T']) st[i] = 1;
		if (i >= k) cnt[s[i - k]]--;
	}
	memset(cnt, 0, sizeof(cnt));
	for (int i = n - 1; i >= 0; i--) {
		cnt[s[i]]++;
		if (cnt['A']) sa[i] = 1;
		if (cnt['C']) sc[i] = 1;
		if (cnt['G']) sg[i] = 1;
		if (cnt['T']) st[i] = 1;
		if (i + k <= n - 1) cnt[s[i + k]]--;
	}
	memset(cnt, 0, sizeof(cnt));
	for (int i = 0; i < m; i++)
		cnt[t[i]]++;
	FFTinit(); memset(ans, true, sizeof(ans));
	for (int i = 0; i < N; i++)
		res[i] = (point) {sa[i] + ta[i], sa[i] - ta[i]};
	FFT(res, 1);
	for (int i = 0; i < N; i++)
		res[i] = res[i] * res[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['A']) > eps) ans[i] = false;
	for (int i = 0; i < N; i++)
		res[i] = (point) {sc[i] + tc[i], sc[i] - tc[i]};
	FFT(res, 1);
	for (int i = 0; i < N; i++)
		res[i] = res[i] * res[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['C']) > eps) ans[i] = false;
	for (int i = 0; i < N; i++)
		res[i] = (point) {sg[i] + tg[i], sg[i] - tg[i]};
	FFT(res, 1);
	for (int i = 0; i < N; i++)
		res[i] = res[i] * res[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['G']) > eps) ans[i] = false;
	for (int i = 0; i < N; i++)
		res[i] = (point) {st[i] + tt[i], st[i] - tt[i]};
	FFT(res, 1);
	for (int i = 0; i < N; i++)
		res[i] = res[i] * res[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['T']) > eps) ans[i] = false;
	int sum = 0;
	for (int i = m - 1; i < n; i++)
		sum += ans[i];
	writeln(sum);
	return 0;
}
/*Ordinary FFT Version*/
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 524288;
const int MAXC = 256;
const double eps = 1e-3;
const double pi = acos(-1);
template <typename T> void chkmax(T &x, T y) {x = max(x, y); }
template <typename T> void chkmin(T &x, T y) {x = min(x, y); } 
template <typename T> void read(T &x) {
	x = 0; int f = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
	for (; isdigit(c); c = getchar()) x = x * 10 + c - '0';
	x *= f;
}
template <typename T> void write(T x) {
	if (x < 0) x = -x, putchar('-');
	if (x > 9) write(x / 10);
	putchar(x % 10 + '0');
}
template <typename T> void writeln(T x) {
	write(x);
	puts("");
}
struct point {double x, y; };
point operator + (point a, point b) {return (point) {a.x + b.x, a.y + b.y}; }
point operator - (point a, point b) {return (point) {a.x - b.x, a.y - b.y}; }
point operator * (point a, point b) {return (point) {a.x * b.x - a.y * b.y, a.x * b.y + a.y * b.x}; }
point operator / (point a, double x) {return (point) {a.x / x, a.y / x}; }
point sa[MAXN], sc[MAXN], sg[MAXN], st[MAXN];
point ta[MAXN], tc[MAXN], tg[MAXN], tt[MAXN];
point res[MAXN]; char s[MAXN], t[MAXN];
int n, m, k, N, Log, cnt[MAXC], home[MAXN];
bool ans[MAXN];
void FFTinit() {
	for (int i = 0; i < N; i++) {
		int tmp = i, ans = 0;
		for (int j = 1; j <= Log; j++) {
			ans <<= 1;
			ans += tmp & 1;
			tmp >>= 1;
		}
		home[i] = ans;
	}
}
void FFT(point *a, int mode) {
	for (int i = 0; i < N; i++)
		if (home[i] < i) swap(a[i], a[home[i]]);
	for (int len = 2; len <= N; len <<= 1) {
		point delta = (point) {cos(2 * pi / len * mode), sin(2 * pi / len * mode)};
		for (int i = 0; i < N; i += len) {
			point now = (point) {1, 0};
			for (int j = i, k = i + len / 2; k < i + len; j++, k++) {
				point tmp = a[j];
				point tnp = a[k] * now;
				a[j] = tmp + tnp;
				a[k] = tmp - tnp;
				now = now * delta;
			}
		}
	}
	if (mode == -1) {
		for (int i = 0; i < N; i++)
			a[i] = a[i] / N;
	}
}
int main() {
	read(n), read(m), read(k);
	scanf("\n%s\n%s", s, t);
	N = 1, Log = 0;
	while (N <= n) {
		N <<= 1;
		Log++;
	}
	reverse(t, t + m);
	for (int i = 0; i < m; i++) {
		if (t[i] == 'A') ta[i] = (point) {1, 0};
		if (t[i] == 'C') tc[i] = (point) {1, 0};
		if (t[i] == 'G') tg[i] = (point) {1, 0};
		if (t[i] == 'T') tt[i] = (point) {1, 0};
	}
	for (int i = 0; i < n; i++) {
		cnt[s[i]]++;
		if (cnt['A']) sa[i] = (point) {1, 0};
		if (cnt['C']) sc[i] = (point) {1, 0};
		if (cnt['G']) sg[i] = (point) {1, 0};
		if (cnt['T']) st[i] = (point) {1, 0};
		if (i >= k) cnt[s[i - k]]--;
	}
	memset(cnt, 0, sizeof(cnt));
	for (int i = n - 1; i >= 0; i--) {
		cnt[s[i]]++;
		if (cnt['A']) sa[i] = (point) {1, 0};
		if (cnt['C']) sc[i] = (point) {1, 0};
		if (cnt['G']) sg[i] = (point) {1, 0};
		if (cnt['T']) st[i] = (point) {1, 0};
		if (i + k <= n - 1) cnt[s[i + k]]--;
	}
	memset(cnt, 0, sizeof(cnt));
	for (int i = 0; i < m; i++)
		cnt[t[i]]++;
	FFTinit(); memset(ans, true, sizeof(ans));
	FFT(sa, 1); FFT(sc, 1); FFT(sg, 1); FFT(st, 1);
	FFT(ta, 1); FFT(tc, 1); FFT(tg, 1); FFT(tt, 1);
	for (int i = 0; i < N; i++)
		res[i] = sa[i] * ta[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['A']) > eps) ans[i] = false;
	for (int i = 0; i < N; i++)
		res[i] = sc[i] * tc[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['C']) > eps) ans[i] = false;
	for (int i = 0; i < N; i++)
		res[i] = sg[i] * tg[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['G']) > eps) ans[i] = false;
	for (int i = 0; i < N; i++)
		res[i] = st[i] * tt[i];
	FFT(res, -1);
	for (int i = m - 1; i < n; i++)
		if (fabs(res[i].x - cnt['T']) > eps) ans[i] = false;
	int sum = 0;
	for (int i = m - 1; i < n; i++)
		sum += ans[i];
	writeln(sum);
	return 0;
}

cz_xuyixuan
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
binarysearch-codeforces
04-01
binarysearch-codeforces
CodeForces:CodeForces解决方案
04-19
CodeForces解决方案 Codeforces是一个托管竞争性编程竞赛的网站。 它是由来自ITMO大学的一组有竞争力的程序员维护的，由Mikhail Mirzayanov领导。 自2013年以来，Codeforces声称在活跃参赛者方面超过Topcoder。 截至...
CodeForces - 528D Fuzzy Search(多项式匹配字符串)
Falcon的博客
07-28 268
题目链接：点击查看 题目大意：给出一个长度为 nnn 的字符串 sss 和一个长度为 mmm 的字符串 ttt，问字符串 sss 有哪些子串可以匹配 ttt。给出一个参数 kkk，TTT 在 SSS 的第 iii 个位置中出现，当且仅当把 TTT 的首字符和 SSS 的第 iii 个字符对齐后，TTT 中的每一个字符能够在 SSS 中找到一个位置偏差不超过 kkk 的相同字符。 需要注意的是，在字符串 sss 中通过偏差匹配的字符可以重复与 ttt 匹配 题目分析：预处理出数组 g[i][j]g[i][j]
Codeforces #528D: Fuzzy Search 题解
IcePrincess_1968's World
04-24 269
首先每个位置是否能被A,T,G,C覆盖到是可以线性预处理的 然后我们可以把四种字符分开考虑，这样母串和匹配串都是一个01串，我们想试图对每个母串中的位置，看从这个位置出发的和匹配串长度一样的子串和匹配串是否相同 尝试考虑后缀数组+lcp的做法，但发现一个问题：母串的子串不一定要和匹配串相同，只要它们或的值和母串的子串一样都行（也就是匹配串是1的地方母串也是1) 然后就不会做了… 下面是...
Codeforces528 D. Fuzzy Search（bitest）
zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz
09-03 270
题意： 给定长度为n的串S和长度为m的串T，还有一个整数k 串只由ACGT组成，问T串在S串中出现了多少次，匹配机制比较特殊： 定义T[i]能匹配上S[j]，当且仅当T[i]在S[j-k,j+k]中出现过，即存在偏移位差k 数据范围：n,m,k<=2e5 ...
CF528D Fuzzy Search [FFT]
FSYo 的博客
07-09 205
传送门 首先预处理一个 0 / 1数组, 表示这个位置能不能与 A / C / G / T 匹配 由于字符集很小, 所以我们可以一个字符一个字符处理 我们将 B 串是当前枚举字符的位置赋成1, 那么两个都是1就可以匹配 于是 FFT 将它们卷起来, 并且将4次处理的多项式相加, 如果某一位系数为 m, 那么每个字符都匹配上了 #include<bits/stdc++.h> ...
Codeforces 528 D. Fuzzy Search (FFT)
hddddh的博客
10-13 187
链接：D. Fuzzy Search 题意： 给出一个母串S和一个模板串T，求模板串在母串中的匹配次数。 匹配时，如果用s[i]匹配t[j]，那么只要s[i-k]-s[i+k]中有字母与t[j]相同即可算作匹配成功。 思路： 对于普通的字符串匹配，我们考虑每一个字符，只要把模板串倒过来，两个串中是该字符的位置设置为 1 ，跑一次 FFT , 求出每一个字符对模板穿移动 i 位的贡献，如果所有字母的贡献 如果等于模板串的长度，答案就+1. 对于此题，如果母串中第 i 位为该字符，那么我们就把 [i - k
CodeforcesPP:Codeforces扩展包
05-06
Codeforces ++ Codeforces扩展包 相关连结 特征 “显示标签”按钮 导航栏下拉菜单使导航更加轻松 mashup / gym问题上的“ Google It”按钮 教程弹出按钮 自定义样式 自动更新排名页面 键盘快捷键（请参见下文） 在...
Codeforces codes_names_Codeforces_
10-02
《Codeforces代码名称解析》 Codeforces是一个全球知名的在线编程竞赛平台，吸引了众多程序员和算法爱好者参与。这里的“codes_names_Codeforces_”标题暗示我们将探讨的是Codeforces平台上一些问题的代码名称。...
【codeforces】528D. Fuzzy Search【FFT】
poursoul
05-07 2741
传送门：【codeforces】528D. Fuzzy Search题目分析：首先，我们先来了解一下FFTFFT求字符串匹配的方法。问题：求BB串在AA串中匹配的次数。解法： 我们用aia_i表示串AA第ii个位置的字符，bjb_j表示串BB第jj个位置的字符。AA串用多项式表示为： A=∑ni=1xi=∑ni=1A(i)\qquad\qquad A = \sum_{i=1}^n x^i=\su
codeforces 528D. Fuzzy Search （FFT优化DP）
clover_hxy的博客
02-23 1207
题目描述传送门题目大意：给出一个母串和一个模板串，求模板串在母串中的匹配次数。 匹配时，如果用s[i]匹配t[j]，那么只要s[i-k]-s[i+k]中有字母与t[j]相同即可算作匹配成功。其中s[i]表示母串的第i位，t[j]表示模板串的第j位。题解如果数据范围小的话，这题就是一个DP f[i][j]=f[i-1][j-1]&mp[i][t[j]]表示母串的第i位匹配到模板串的第j位是否可以匹
codeforces 528D Fuzzy Search
我要吃熊猫
05-31 383
FFT
CodeForces 528D Fuzzy Search（FFT）
Bfk的博客
02-05 552
CodeForces 528D Fuzzy Search原题地址：http://codeforces.com/problemset/problem/528/D题意： 给给定两个字符串 S，T（字符集大小为4：A ,G ,C ,T），给定阀值 k， 定义T串在S串某位置匹配，当且仅当T串任意位置的这个字符所对应的母串的位置的左右k个字符之内至少有一个与这个字符相同的。 求T串在S串中的匹配次数。
【CF528D】Fuzzy Search
洛水·锦依卫's Blog
09-08 258
Problem Description 你有一个长度为 \(n\) 的串 \(S\)，以及长度为 \(m\) 的串 \(T\)。 现给定一个数 \(k\) ，我们说 \(T\) 在 \(S\) 的位置 \(i\) 匹配上，当且仅当对于每一个 \(1\le a\le m\) ，都有一个位置 \(1\le b\le n\) 满足 \(|(i+a-1)-b|\le k\) ，且 \(S_b=T...
codeforces 528D Fuzzy Search FFT
make_it_for_good的博客
12-13 390
单调处理出第一个串往两边K个能覆盖的字符。 然后对四种字符每种字符做一遍FFT。 对于第一个串，如果这个位置不能匹配当前字符，那该位置为1。 对于第二个串，如果这个位置为当前字符，那该位置为1。 把第二个串反过来和第一个串跑FFT。 答案是在所有FFT中都为0的合法位置个数。#include <bits/stdc++.h> using namespace std; #define N 21
Codeforces 528D
Mr_Xujh的博客
10-21 769
题意：给定一个主串和一个模式（都只含有ATCG）,求模式在主串中出现的次数 但是匹配的时候有一个偏移量K，即在用t[i]匹配s[j]的时候  只要s[j-k]---s[j+k]中有字符t[i]就算匹配； 这里用了FFT 参考：http://blog.csdn.net/u013368721/article/details/45565729 #include #include
【CodeForces】528D Fuzzy Search
include / Hdhd
04-21 248
题目描述 Leonid works for a small and promising start-up that works on decoding the human genome. His duties include solving complex problems of finding certain patterns in long strings consisting of let...
[CF528D]Fuzzy Search
abcyan1235的博客
04-06 145
luogu sol 这种字符串匹配的问题显然可以把一个串\(reverse\)过来然后用\(FFT\)做吧。 对每种字母分开考虑，设两个多项式\(A(x),B(x)\)，其中 \[A(x)=\sum_{i=0}^{n-1}[区间[i-k,i+k]内存在该种字符]x^i\] \[B(x)=\sum_{i=0}^{n-1}[t[i]为该种字符]x^i\] 然后两个多项式卷一下就是这种字符在第...
codeforces
最新发布
10-01
Codeforces是一个著名的在线编程竞赛平台，提供了大量的算法和编程题目供程序员刷题。在你提供的引用中，这些引用包含了一些Codeforces的比赛和问题的名称。其中，CodeForces 984A，CodeForces 1132B，CodeForces 1015C是排序题目的名称。Codeforces Round #301 (Div. 2)是一场比赛的名称，其中的问题是Combination Lock。Codeforces-1672 C: Unequal Array是另一个问题的名称。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值