CODEVS——T 1404 字符串匹配

http://codevs.cn/problem/1404/ 

时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 大师 Master

题解

 查看运行结果

 

 

题目描述 Description

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

输入描述 Input Description

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

输出描述 Output Description

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

样例输入 Sample Input

6 2 2
aabcde
ab
0
2

样例输出 Sample Output

4
1

数据范围及提示 Data Size & Hint

各个测试点1s

kmp得出ans[],然后考虑将串分解，，倒序累加ans[next[i]],,此时的ans[i]是>=i位的数量，所以最后还要处理为ans[i]-=ans[i+1]

#include <algorithm>
#include <cstring>
#include <cstdio>

using namespace std;

const int N(200000+5);
int n,m,k,lb,la;
int p[N],ans[N];
char a[N],b[N];

inline void Get_next()
{
    for(int i=2,j=0;i<=lb;p[i++]=j)
    {
        for(;b[i]!=b[j+1]&&j>0;) j=p[j];
        if(b[i]==b[j+1]) j++;
    }
}
inline void kmp()
{
    for(int i=1,j=0;i<=la;i++)
    {
        for(;a[i]!=b[j+1]&&j>0;) j=p[j];
        if(a[i]==b[j+1]) j++;
        ans[j]++;
    }
}

int main()
{
    scanf("%d%d%d%s%s",&n,&m,&k,a+1,b+1);
    la=strlen(a+1); lb=strlen(b+1);
    Get_next(); kmp();
    for(int i=lb;i>0;i--) ans[p[i]]+=ans[i];
    for(int i=0;i<lb;i++) ans[i]-=ans[i+1];
    for(int pos;k--;)
    {
        scanf("%d",&pos);
        printf("%d
",ans[pos]);
    }
    return 0;
}