JZOJ 4936 【NOIP2017GDKOI模拟1.12】c

Description

给出一个长度为n的01串S。
两个01串相似要满足两个条件：1、长度相同。2、含有1或含有0的数量相同。
现在给出若干个01串进行询问，对于每一个被询问的01串，求S有多少个子串与每一个被询问的01串相似。

Data Constraint

n<=2* 105，每一个被询问串非空，且所有的被询问串的长度总和不超过2*105

Solution

一次匹配的复杂度为O(n)。
若每一次询问串的长度都大于P−−√（P=2*105）,这种情况下总匹配的时间复杂度即为O(nP−−√)。
但询问串的长度不一定都大于P−−√，因此可以先预处理出所有长度小于P−−√的情况，预处理出F数组，Fi,j表示长度为i，1的数量为j的情况数，预处理的时间复杂度为O(n P−−√)，因此总的时间复杂度为O(n\sqrt P$)，勉强可以过掉。

Code(Pascal)

var    i,k,l,j,o,p,q,ans:longint;    f:array[0..500,0..1000] of longint;    zj:array[0..250000] of longint;    st:ansistring;begin    readln(st);    p:=length(st);    for i:=1 to p do    if st[i]='1' then zj[i]:=zj[i-1]+1 else zj[i]:=zj[i-1];    o:=trunc(sqrt(p));    for i:=1 to o do    for l:=i to p do    inc(f[i,zj[l]-zj[l-i]]);    readln(q);    for i:=1 to q do    begin        readln(st);        k:=length(st);  j:=0;        for l:=1 to k do  if st[l]='1' then inc(j);        if k<=o then writeln(f[k,j])        else        begin            ans:=0;            for l:=k to p do if zj[l]-zj[l-k]=j then inc(ans);            writeln(ans);        end;    end;end.