codeforces1016B 字符串区间查找子串(数组前缀和防TLE)

题目链接:http://codeforces.com/problemset/problem/1016/B

题意是给你给两个字符串s,t,之后给出s的q个区间,对每个区间[l,r]询问s的这个区间中存在多少个子串与t相同

一开始的思路是每给出一个区间[L,R]。便在s相应的区间用KMP进行查找t出现的次数,但是大佬们说自己暴力过了这题。而我的暴力一直TLE。后来在排行榜上找了份大神的代码,原来他们的暴力是:预处理是用一个数组sum记录字符串中t出现的次数。sum[i]记录s串中前i个字符中t串(头部在i)出现的总次数。查找时直接利用端点的前缀和相减

思考:前缀和在很多题中的确可以节约重复的操作,例如查找区间[1,1000],后又查找[2,1001]。则[2,1000]被重复查找,而利用前缀和便可以解决这个问题,一次查找记录了所有的信息。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <math.h>
#include <string>
#include <time.h>
#include <algorithm>
#include <set>
#include <stack>
#include <queue>
using namespace std;

char a[1005];
char b[1005];
int t[1005];
int sum[1005];

int main()
{
    memset(t,0,sizeof(t));
    int p, n, m;
    cin>>n>>m>>p;
    cin>>a>>b;

    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(a[i+j]!=b[j])
                break;
            else if(j==m-1)
                //s串中t串出现的位置i(头)
                t[i]=1;
        }
    }

    sum[0]=t[0];
    for(int i=1;i<n;i++)
        //s串前i个字符中t串出现的次数是sum[i]
        sum[i]=sum[i-1]+t[i];

    while(p--)
    {
        int l, r;
        scanf("%d%d",&l, &r);
        l--, r--;
        if(n<m||r-l+1<m)
            cout<<0<<endl;
        else
        //查找区间[l, r]只需要两个区间端点的t串出现次数相减
            cout<<sum[r-m+1]-sum[l-1]<<endl;
    }

    return 0;
}
全部评论

相关推荐

程序员小白条:投太少了,多投点吧,二本就海投,然后简历上加点奖项或者四六级之类的,别管有没有用,另外最好搞下个人博客,定期输出一些文章和学习总结,也可以去github参与一下开源项目提一些PR,总会有中小公司看的上的
点赞 评论 收藏
分享
头像
11-21 11:39
四川大学 Java
是红鸢啊:忘了还没结束,还有字节的5k 违约金
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务