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;
}
全部评论

相关推荐

年少的我,一直都很期待别人告诉我的“去大城市见世面”,高考没能考出河南省,甚至上的大学离市区很远,公交地铁两个小时起步。当时我眼中的“去大城市见世面”,就是去大城市工作,看高楼大厦、看车水马龙、游览各种景点、赚很多很多钱、去酒吧去好吃的餐厅、做着高大上的工作、见牛B的人。大一时,我偶然知道了字节的稀土开发者大会,看到了很多大厂背景的人,又因为对大城市的向往,家庭经济很一般甚至有点拮据的我,跟朋友借了几十块钱,买了从郑州到北京的硬座,一晚上6小时到了北京。我确实见到了北京火车站干净的厕所、朝阳区有序的交通、骑着车路过了天安门,那场开发者大会,我也确实见到了学习视频里的人,但是由于自身知识储备的不...
我推的MK:我刚来到大城市的时候就是完全祛媚的,狭小出租屋,天价的房租和恐怖的晚高峰,但是大城市依旧给我带来了很多不一样的东西,比如实习机会,比如逛不尽的商场,比如最前沿的艺术展,比如演唱会和比赛,每当我想要遇见新事物的时候都能够很轻易地在大城市里找到,每当我想要打卡的时候发现坐一个小时地铁都不算多远……大城市就像一片广阔的海洋,我们待在水洼里并不觉得怎样,但当你无论如何眺望都看不到广袤的边界时,你会骤然意识到大城市的意义。
点赞 评论 收藏
分享
纸鹰:对他说:“你好,我是百度JAVA。”
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务