tyvj 1068 STR

STR  tyvj 1068

 

描述 Description

给你两个串A，B，可以得到从A的任意位开始的子串和B匹配的长度。
给定K个询问，对于每个询问给定一个x，求出匹配长度恰为x的位置有多少个。
N，M，K<=200000

 

输入格式 Input Format

第一行三个数 N，M，K，表示A的长度、B的长度和询问数。
第二行为串A。
第三行为串B。
接下来K行，每行1个数X。

 

输出格式 Output Format

对于每个询问输出一个数。

 

样例输入 Sample Input

6 2 2

aabcde

ab

0

2

 

样例输出 Sample Output 

4

1

 

时间限制 Time Limitation

各个测试点1s

 

用KMP，匹配过程中记录在某一长度下的位置个数，最后统计答案即可

 

 

var
  n,m,k,i,j,x:longint;
  a,b:ansistring;
  p,l,ans:array[0..200001] of longint;
begin
  assign(input,'p1068.in');
  reset(input);
  assign(output,'p1068.out');
  rewrite(output);
  
  readln(n,m,k);
  readln(a);
  readln(b);
  j:=0;
  for i:=2 to m do
    begin
      while (j>0)and(b[j+1]<>b[i]) do j:=p[j];
      if b[j+1]=b[i] then inc(j);
      p[i]:=j;
    end;
  j:=0;
  for i:=1 to n do
    begin
      while (j>0)and(b[j+1]<>a[i]) do j:=p[j];
      if b[j+1]=a[i] then inc(j);
      l[i]:=j;  //记录长度
      if j=m then j:=p[j];
    end;
  for i:=1 to n do inc(ans[l[i]]);
  for i:=m downto 1 do inc(ans[p[i]],ans[i]); //能匹配i长度，一定也能匹配p[i]长度
  for i:=1 to k do
    begin
      readln(x);
      writeln(ans[x]-ans[x+1]); //ans[x]表示x以上(包括x)的位置数，因此减去ans[x+1]就是答案
    end;

  close(input);
  close(output);
end.