[NOI2018]你的名字

89 篇文章 0 订阅
72 篇文章 0 订阅

https://www.luogu.org/problemnew/show/P4770

sam根本不理解.jpg,自己想只会搞一个n根号的做询问1~n的垃圾68分做法

真实做法我是对着洛谷little_gift大佬的题解学的,写的很好。

考虑68分里把每个T扔到S的sam上跑匹配,跑到T的下标位置x时它的意义是对T【1~x】这个前缀,从S的所有子串里,找能和 该前缀的后缀 匹配长度最大 的某个子串(即从末尾到前匹配)和它匹配的长度len,也就是说S的所有子串中,从后到前匹配的话,最长可以和T【1~x】这个串匹配为T【x-len+1,x】,假设把这个东西记为mxl【x】。求出这个就很好搞了,再对T建一个sam,建的时候维护节点代表的串(right集合)的一个出现位置,记为tag(其实只要建的时候把下标传入,np类节点的tag就是pos,分裂nt的时候tag也拷贝t的tag就好了),那么因为对于T的一个节点i,它代表了长为【l,r】的一些T的本质不同子串,它在S的子串中会被匹配后缀长度则是【1,mxl【tag【i】】】,所以区间【l,r】除去与这东西交的部分就是答案所求。

那么现在问题就是怎么对S【l...r】求之前那个mxl数组,其实就是在匹配维护len的时候,不仅在失配时往回跳,如果能往下走,还要在走到下一个节点的right集合(记为{a1,a2,a3...})存在一个ai,使得【ai-(len+1)+1,ai】是【l,r】的子集,如果不是,就缩短len的大小,缩到长度不属于这个点了就往回跳。考虑如何查询一个节点的right集合在某个区间是否存在元素,发现可以线段树合并,刚开始建sam时把pos传入丢到np结点的线段树里,然后拓扑排序合并上来,需要注意这里合并时不能按平时写法写,合并的每一步假设要变更某一个结点的孩子都要新开结点代替它来执行(类似可持久化),以保证合并后子节点的线段树不会因为父亲的线段树修改而修改(刚开始有点担心空间,但考虑一下跟mg次数同阶所以没有问题)。

代码:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=2e6+100;
const int M=5e5+100;


int n,mxl[N],tax[N],tnum,rnk[N];
char S[N];

namespace Seg{
    int ls[M*50],rs[M*50],tot;
    int mg(int x,int y)
    {
        if(!x||!y)return x|y;
        int nw=++tot;
        ls[nw]=mg(ls[x],ls[y]);
        rs[nw]=mg(rs[x],rs[y]);
        return nw;
    }
    void ins(int &x,int to,int l=1,int r=M)
    {
        if(!x)x=++tot;
        if(l==r)return;
        int mid=(l+r)>>1;
        if(to<=mid)ins(ls[x],to,l,mid);
        else ins(rs[x],to,mid+1,r);
    }
    bool qry(int x,int tl,int tr,int l=1,int r=M)
    {
        if(!x||tl>tr)return 0;
        if(tl<=l&&r<=tr)return 1;
        int mid=(l+r)>>1;bool res=0;
        if(tl<=mid)res|=qry(ls[x],tl,tr,l,mid);
        if(!res&&tr>mid)res|=qry(rs[x],tl,tr,mid+1,r);
        return res;
    }
}

struct Sam{
    int tot,las,mx[N],nxt[N][26],pre[N],tag[N],rt[N];
    Sam(){tot=las=1;}
    void clr()
    {
        for(int i=1;i<=tot;i++)
            memset(nxt[i],0,sizeof nxt[i]),pre[i]=mx[i]=tag[i]=0;
        tot=las=1;
    }
    void app(int x,int ps,int op)
    {
        int p,np,t,nt;
        p=las,np=las=++tot,mx[np]=mx[p]+1,tag[np]=ps;
        if(op==1)Seg::ins(rt[np],ps);
        while(p&&!nxt[p][x])nxt[p][x]=np,p=pre[p];
        if(!p){pre[np]=1;return;}
        t=nxt[p][x];
        if(mx[t]==mx[p]+1){pre[np]=t;return;}
        nt=++tot;
        mx[nt]=mx[p]+1,pre[nt]=pre[t],tag[nt]=tag[t],pre[t]=pre[np]=nt;
        memcpy(nxt[nt],nxt[t],sizeof nxt[t]);
        for(;p&&nxt[p][x]==t;p=pre[p])nxt[p][x]=nt;
    }
    void mg_seg()
    {
        tnum=0;
        for(int i=1;i<=tot;i++)
            tax[mx[i]]++;
        for(int i=1;i<=tot;i++)
            tax[i]+=tax[i-1];
        for(int i=1;i<=tot;i++)
            rnk[tax[mx[i]]--]=i;
        for(int i=tot,x;i>=1;i--)
        {
            x=rnk[i];
            rt[pre[x]]=Seg::mg(rt[pre[x]],rt[x]);
        }
    }
    void pp(char *S,int ld,int rd)
    {
        int n=strlen(S+1),nw=1,len=0,x,tx;
        for(int i=1;i<=n;i++)
        {
            x=(int)(S[i]-'a');
            for(;nw;)
            {
                if(!nxt[nw][x])
                {
                    nw=pre[nw],len=mx[nw];
                    continue;
                }
                tx=nxt[nw][x];
                if(Seg::qry(rt[tx],ld+len,rd)==1){break;}
                if(len>0)
                {
                    --len;
                    if(len==mx[pre[nw]])nw=pre[nw];
                }
                else nw=0;
            }
            if(nw==0){nw=1;}
            else
            {
                nw=nxt[nw][x];
                ++len;
            }
            mxl[i]=len;
        }
    }
    void sol()
    {
        ll ans=0;
        for(int i=2,ld,rd;i<=tot;i++)
        {
            ld=mx[pre[i]]+1,rd=mx[i];
            ld=max(ld,mxl[tag[i]]+1);
            ans+=max(0,rd-ld+1);
        }
        printf("%lld\n",ans);
    }
}sam1,sam2;

int main()
{
    scanf("%s",S+1);
    n=strlen(S+1);
    for(int i=1;i<=n;i++)
        sam1.app(S[i]-'a',i,1);
    sam1.mg_seg();
    int Q,ld,rd;
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%s",S+1);
        scanf("%d%d",&ld,&rd);
        n=strlen(S+1);
        sam2.clr();
        for(int i=1;i<=n;i++)
            sam2.app(S[i]-'a',i,2);
        sam1.pp(S,ld,rd),sam2.sol();
    }
}

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值