码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces 666E Forensic Examination SAM+权值线段树

时间:2018-05-18 18:13:24      阅读:211      评论:0      收藏:0      [点我收藏+]

标签:子串   opera   stdin   fine   eof   线段   %s   bug   root   

第一次做这种$SAM$带权值线段树合并的题 然而$zjq$神犇看完题一顿狂码就做出来了 $Orz$

首先把所有串当成一个串建$SAM$ 我们对$SAM$上每个点 建一棵权值线段树 每个叶子节点表示一个匹配串能到达这个点的子串个数 这样我们对最后的$SAM$的权值线段树按$parent$树合并 询问的时候找到对应的$SAM$上的权值线段树直接查询就好了

具体的操作看代码吧~

#include<bits/stdc++.h>
using namespace std;
#define FO(x) {freopen(#x".in","r",stdin);freopen(#x".out","w",stdout);}
#define pa pair<int,int>
#define mod 1000000007
#define ll long long
#define mk make_pair
#define pb push_back
#define fi first
#define se second
#define cl(x) memset(x,0,sizeof x)
#ifdef Devil_Gary
#define bug(x) cout<<(#x)<<" "<<(x)<<endl
#define debug(...) fprintf(stderr, __VA_ARGS__)
#else
#define bug(x)
#define debug(...)
#endif
const int INF = 0x7fffffff;
const int N=1230010;
/*
char *TT,*mo,but[(1<<15)+2];
#define getchar() ((TT==mo&&(mo=(TT=but)+fread(but,1,1<<15,stdin),TT==mo))?-1:*TT++)//*/
inline int read(){
    int x=0,rev=0,ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')rev=1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return rev?-x:x;
}
pa operator * (const pa&A,const pa&B){return A.fi==B.fi?(A.se<B.se?A:B):(A.fi>B.fi?A:B);}
pa operator + (const pa&A,const pa&B){return mk(A.fi+B.fi,A.se);}
pa t[N*23];
char s[N];
int n,m,Q,c[N][27],par[N],len[N],fa[N][21],ls[N*23],rs[N*23],h[N],rt=1,sz=1,last=1,root[N],w[N],b[N],id; 
void extend(int x){
    int np=++sz,p=last;
    len[np]=len[p]+1,last=np;
    for(;p&&!c[p][x];p=par[p]) c[p][x]=np;
    if(!p) par[np]=rt;
    else{
        int q=c[p][x];
        if(len[p]+1==len[q]) par[np]=q;
        else{
            int nq=++sz;
            len[nq]=len[p]+1,par[nq]=par[q],par[q]=par[np]=nq;
            memcpy(c[nq],c[q],sizeof c[nq]);
            for(;p&&c[p][x]==q;p=par[p]) c[p][x]=nq;
        }
    }
}
void modify(int&pos,int l,int r,int x){
    if(!pos) pos=++id; 
    if(l==r){
        t[pos].fi++,t[pos].se=l;
//      cout<<t[pos].fi<<" "<<t[pos].se<<endl;
        return;
    }
    int mid=l+r>>1;
    if(x<=mid) modify(ls[pos],l,mid,x);
    else modify(rs[pos],mid+1,r,x);
    t[pos]=t[ls[pos]]*t[rs[pos]];
}
inline int merge(int x,int y,int l,int r)
{
    if (!x||!y) return x|y; 
    int z=++id;
    if (l==r) {
        t[z]=t[x]+t[y];
        return z;
    }
    int mid=(l+r)>>1;
    ls[z]=merge(ls[x],ls[y],l,mid),rs[z]=merge(rs[x],rs[y],mid+1,r);
    t[z]=t[ls[z]]*t[rs[z]];
    return z;
}
pa Query(int pos,int l,int r,int nl,int nr){
    if(!pos) return mk(0,0);
    if(nl<=l&&r<=nr) return t[pos];
    int mid=l+r>>1;
    if(nr<=mid) return Query(ls[pos],l,mid,nl,nr);
    else if(nl>mid) return Query(rs[pos],mid+1,r,nl,nr);
    else return Query(ls[pos],l,mid,nl,nr)*Query(rs[pos],mid+1,r,nl,nr);
}
void topsort(){
    for(int i=1;i<=sz;i++) w[len[i]]++;
    for(int i=1;i<=sz;i++) w[i]+=w[i-1];
    for(int i=1;i<=sz;i++) b[w[len[i]]--]=i;
    for(int i=sz;i;i--){
        int j=b[i];
        if(par[j]) root[par[j]]=merge(root[par[j]],root[j],1,n);
    }
}
void get_fa(){
    for(int i=1;i<=sz;i++) fa[i][0]=par[i];
    for(int i=1;i<=20;i++) for(int j=1;j<=sz;j++) fa[j][i]=fa[fa[j][i-1]][i-1];
}
int get(int l,int r){
    int L=r-l+1,x=h[r];
    for(int i=20;~i;i--) if(len[fa[x][i]]>=L) x=fa[x][i];
//  -len[par[x]]-len[par[fa[x][i]]]
    return x;
}
int main(){
#ifdef Devil_Gary
    freopen("in.txt","r",stdin);
#endif
    scanf("%s",s+1),m=strlen(s+1);
    for(int i=1;i<=m;i++) extend(s[i]-'a'),h[i]=last;
    n=read();
    for(int i=1;i<=n;i++){
        scanf("%s",s+1),m=strlen(s+1),extend(26);
        for(int j=1;j<=m;j++) extend(s[j]-'a'),modify(root[last],1,n,i);
    }
//  bug(sz);
    topsort(),get_fa();
    for(Q=read();Q;Q--){
        int l=read(),r=read(),pl=read(),pr=read(),x=get(pl,pr);
        pa ans=Query(root[x],1,n,l,r);
        if(!ans.fi) printf("%d 0\n",l);
        else printf("%d %d\n",ans.se,ans.fi);
    } 
}

Codeforces 666E Forensic Examination SAM+权值线段树

标签:子串   opera   stdin   fine   eof   线段   %s   bug   root   

原文地址:https://www.cnblogs.com/devil-gary/p/9057102.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!