YbtOj#20072. 「NOIP2020 模拟赛 B 组 Day6」相似子串

$Result$

暴力：

正解：

---

$Hyperlink$

http://noip.ybtoj.com.cn/contest/105/problem/2

---

$Description$

给定一个01串$s$，$T$组询问，每次询问给定一个串$t$，询问$s$有多少个长度和$t$一样的子串，使得区间的1的和相同

数据范围：$|s|\le 2\times 10^5,\sum |t|\le 2\times 10^5,T\le 2\times 10^5$

---

$Solution$

暴力能过。。。。当然我们要想更加优秀的做法

由于$t$的长度和是有限的，这就引出了优秀的性质，即

1. $t$的长度的种类不会超过$\sqrt{n}$种
2. $|t|>\sqrt{n}$ 的至多有$\sqrt{n}$个

根据性质1，我们可以将询问离线，同种长度的一起计算，时间复杂度：$O(n\sqrt{n})$

根据性质2，我们可以长度$\le \sqrt{n}$的答案提前预处理出来，具体地，设$f_{i,j}$表示长度为$i$和为$j$的串的个数，预处理是$O(n\sqrt{n})$的，然后对于$|t|\le \sqrt{n}$的直接$O(1)$输出，大于$\sqrt{n}$的直接跑，因为最多跑$\sqrt{n}$次，所以是可以过得，总的时间复杂度也是$O(n\sqrt{n})$

性质1的话是离线的，性质2的话是在线的，代码采用的是性质2

---

$Code$

暴力

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#define LL long long
#define N 200010
using namespace std;char s[N],st[N];
int n,S[N],m,sum,res,len;
inline LL read()
{
	LL d=1,f=0;char c;
	while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
signed main()
{
	freopen("similar.in","r",stdin);
	freopen("similar.out","w",stdout);
	scanf("%s",s);len=strlen(s);
	for(register int i=0;i<len;i++) S[i+1]=S[i]+(s[i]==49);
	n=read();
	while(n--)
	{
		scanf("%s",st);m=strlen(st);sum=0;res=0;
		for(register int i=0;i<m;i++) sum+=st[i]==49;
		for(register int i=0;i<=len-m+1;i++) if(S[i+m]-S[i]==sum) res++;
		printf("%d\n",res);
	}
}

根号分治

#include<cmath>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#define LL long long
#define N 200010
using namespace std;char s[N],st[N];
int n,S[N],m,sum,res,len,f[450][450],T;
inline LL read()
{
	LL d=1,f=0;char c;
	while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
signed main()
{
	freopen("similar.in","r",stdin);
	freopen("similar.out","w",stdout);
	scanf("%s",s);len=strlen(s);T=sqrt(len);
	for(register int i=0;i<len;i++) S[i+1]=S[i]+(s[i]==49);
	for(register int i=1;i<=T;i++)
	 for(register int j=0;j+i<=len;j++)
	  f[i][S[j+i]-S[j]]++;
	n=read();
	while(n--)
	{
		scanf("%s",st);m=strlen(st);sum=0;
		for(register int i=0;i<m;i++) sum+=st[i]==49;
		if(m>T)
		{
			res=0;
			for(register int i=0;i<=len-m+1;i++) if(S[i+m]-S[i]==sum) res++;
			printf("%d\n",res);
		}
		else printf("%d\n",f[m][sum]);
	}
}