Codeforces 1252D Find String in a Grid

题意

给一个矩形字符串,题问很多串,问每个匹配的次数
匹配的定义是,向右走然后向下走,经过的字符接在一起和提问串一样
比如
在这里插入图片描述
匹配了 ′ A B C ′ 'ABC' ABC , 单独水平或竖直匹配也是可以的(不一定要有拐点)

题解

题解的做法很复杂,用后缀数组搞得很麻烦,然后在vj上找到了一个简单的做法
AC自动机
枚举横行,枚举拐点,然后暴力向下走,在自动机上跑,复杂度 O ( n 3 ) O(n^3) O(n3)

需要注意的是,会重复计算!!!
在这里插入图片描述
如图,红色的是匹配成功,但是,在上面的5个拐点下来的时候,都会被统计一次,所以,要加一个处理,竖直匹配的,只算到第一行的拐点下来的,具体见代码。

代码
#include<bits/stdc++.h>
#define N 200010
#define INF 0x3f3f3f3f
#define eps 1e-8
#define pi 3.141592653589793
#define mod 998244353
#define P 1000000007
#define LL long long
#define pb push_back
#define fi first
#define se second
#define cl clear
#define si size
#define lb lower_bound
#define ub upper_bound
#define bug(x) cerr<<#x<<"      :   "<<x<<endl
#define mem(x,y) memset(x,0,sizeof(int)*(y+3))
#define sc(x) scanf("%d",&x)
#define scc(x,y) scanf("%d%d",&x,&y)
#define sccc(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
typedef pair<int,int> pp;
struct Trie{
    int t[N][26],fail[N],cnt[N],in[N],q[N],h,tail,tot;
    void init(){ tot=0;
        memset(fail,0,sizeof fail);
        memset(t,0,sizeof t);
    }
    int ins(char* x){
        int i=0;
        for(int j=0;x[j];j++){
            int c=x[j]-'A';
            if(!t[i][c]) t[i][c]=++tot;
            i=t[i][c];
        }
        return i;
    }
    void get_fail(){
        h=1,tail=0;
        for (int i=0;i<26;i++) if (t[0][i]) q[++tail]=t[0][i];
        while(h<=tail) {
            int i=q[h++];
            for(int j=0;j<26;j++) {
                if (t[i][j]){
                    fail[t[i][j]]=t[fail[i]][j];
                    q[++tail]=t[i][j];
                }else
                t[i][j]=t[fail[i]][j];
            } 
        }
    }
    void solve(){
        for(int i=tail;i>0;i--) cnt[fail[q[i]]]+=cnt[q[i]];
    }
}T;

char x[510][510];
char s[N];
int id[N];
int main(int argc, char const *argv[])
{
    int n,m,q;
    sccc(n,m,q);
    T.init();
    for(int i=1;i<=n;i++) scanf("%s",x[i]+1);
    for(int i=1;i<=q;i++){
        scanf("%s",s);
        id[i]=T.ins(s);
    }
    T.get_fail();
    for(int i=1;i<=n;i++){
        int p=0;
        for(int j=1;j<=m;j++){
            int t=p,r=0;
            for(int k=i;k<=n;k++){
                r=T.t[r][x[k][j]-'A'];
                t=T.t[t][x[k][j]-'A'];
                T.cnt[t]++;
                T.cnt[r]--;
                if (i==1) T.cnt[r]++;
            }
            p=T.t[p][x[i][j]-'A'];
        }
    }
    T.solve();
    for(int i=1;i<=q;i++) printf("%d\n",T.cnt[id[i]]);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值