UVA 10679 I Love Strings!!(后缀数组)

第一次做后缀数组的题,先处理文本串,整理出字典序然后对比模式串就可以了,感觉预处理对象和AC自动机换了一下。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100000+5;
struct SuffixArray{
    char s[maxn];
    int sa[maxn];
    int t[maxn],t2[maxn],c[maxn],x[maxn],n;
    void clear() {n=0;memset(c,0,sizeof(c));}
    
    void build_sa(int m)
    {
        int *x=t,*y=t2;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[i]=s[i]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=0;i<n;i++) sa[--c[x[i]]]=i;
        for(int k=1;k<=n;k<<=1)
        {
            int p=0;
            for(int i=n-k;i<n;i++) y[p++]=i;
            for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
            for(int i=0;i<m;i++) c[i]=0;
            for(int i=0;i<n;i++) c[x[y[i]]]++;
            for(int i=0;i<m;i++) c[i]+=c[i-1];
            for(int i=0;i<n;i++) sa[--c[x[y[i]]]]=y[i];
            swap(x,y);
            p=1;x[sa[0]]=0;
            for(int i=1;i<n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
            if(p>=n) break;
            m=p;
        }
    }
    
    int cmp_suffix(char *P,int p,int m)
    {
        return strncmp(P,s+sa[p],m);
    }
    
    int find(char *P)
    {
        int m=(int)strlen(P);
        if(cmp_suffix(P,0,m)<0) return -1;
        if(cmp_suffix(P,n-1,m)>0) return -1;
        int L=0,R=n-1;
        while(L<=R)
        {
            int M=(L+R)/2;
            int res=cmp_suffix(P,M,m);
            if(!res) return M;
            if(res>0) L=M+1;else R=M-1;
        }
        return -1;
    }
};

SuffixArray sa;
#define ss(x) scanf("%d",&x)
int main()
{
    int t;ss(t);
    while(t--)
    {
        sa.clear();
        scanf("%s",sa.s);
        sa.n=(int)strlen(sa.s)+1;
        int q;ss(q);
        while(q--)
        {
            char str[1005];
            scanf("%s",str);
            if(sa.find(str)>=0) printf("y\n");else printf("n\n");
        }
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值