bzoj2754 后缀数组

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2754
大意:给n个串,每个串分为2部分(姓和名)。串由(<=10000)的数字组成。再给m个串,求这m个串分别是n个串子串的个数,以及n个串分别包含m个串的个数。

感觉自己最近真的是极度拖延症。并且代码能力也是boom!boom!boom!每次看到难打的题都是只想想就扔一边了。(这是退役的前兆吗。。。)

回归正题,直接将所有串连起来(要注意分隔符要不同)。然后后缀数组+暴力判断。(听说很难卡的)。
(还有种AC自动机的方法,好像也是很暴力)

#include<iostream> 
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,num,Mark[500050],s[500050],pos[500005],Len[500050];
int rank[500050],sa[500050],w[500040],hi[500050];
int vis[500050],ans[500050];
inline bool In_equal(int *x,int i,int j,int k)
{
    int tx = (i+k<num) ? x[i+k] : -1;
    int ty = (j+k<num) ? x[j+k] : -1;
    return (x[i]==x[j]&&tx==ty);
}
inline void Make_Sa() 
{
    int *x=rank,*y=hi,r=10013,yn;
    for (int i=0;i<r;++i) w[i]=0;
    for (int i=0;i<num;++i) ++w[s[i]];
    for (int i=1;i<r;++i) w[i]+=w[i-1];
    for (int i=num-1;i>=0;--i) sa[--w[s[i]]]=i;
    x[sa[0]]=0; r=1;
    for (int i=1;i<num;++i) 
        x[sa[i]]= (s[sa[i]]==s[sa[i-1]]) ? r-1 : r++;
    for (int k=1;r<num;k<<=1) {
        yn=0;
        for (int i=num-k;i<num;++i) y[yn++]=i;
        for (int i=0;i<num;++i)
            if (sa[i]>=k) y[yn++]=sa[i]-k;
        for (int i=0;i<r;++i) w[i]=0;
        for (int i=0;i<num;++i) ++w[x[y[i]]];
        for (int i=1;i<r;++i) w[i]+=w[i-1];
        for (int i=num-1;i>=0;--i) sa[--w[x[y[i]]]]=y[i];
        swap(x,y); r=1; x[sa[0]]=0;
        for (int i=1;i<num;++i)
            x[sa[i]]= In_equal(y,sa[i],sa[i-1],k) ? r-1 : r++;
    }
    if (x!=rank) copy(x,x+num,rank);
    hi[0]=0;
    for (int i=0,k=0,j;i<num;++i) {
        if (k) --k;
        if (!rank[i]) continue;
        j=sa[rank[i]-1];
        while (i+k<num&&j+k<num&&s[i+k]==s[j+k]) ++k;
        hi[rank[i]]=k;
    }
}
inline void Work()
{
    int l,r,tmp;
    for (int i=1;i<=m;++i) {
        l=r=rank[pos[i]];
        while (hi[l]>=Len[i]) --l;
        while (hi[r]>=Len[i]) ++r;
        --r; tmp=0;
        for (int j=l;j<=r;++j) 
            if (Mark[sa[j]]&&vis[Mark[sa[j]]]!=i) {
                vis[Mark[sa[j]]]=i; ++tmp;
                ++ans[Mark[sa[j]]];
            }
        printf("%d\n",tmp);
    }
    for (int i=1;i<=n;++i) {
        printf("%d",ans[i]);
        if (i<n) putchar(' ');
        else printf("\n");
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int k,x;
    for (int i=1;i<=n;++i) {
        scanf("%d",&k);
        while (k--) {
            scanf("%d",&x);
            Mark[num]=i;
            s[num++]=x;
        }
        s[num++]=10001;
        scanf("%d",&k);
        while (k--) {
            scanf("%d",&x);
            Mark[num]=i;
            s[num++]=x;
        }
        s[num++]=10001;
    }
    for (int i=1;i<=m;++i) {
        scanf("%d",&k);
        Len[i]=k;   
        pos[i]=num;
        while (k--) {
            scanf("%d",&x);
            s[num++]=x;
        }
        s[num++]=10011;
    }
    Make_Sa();
    Work();
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值