CF547E Mike and Friends

upd:感谢评论里@spacevortex 指正 暴力链加复杂度是$O(\sum |s|)$的 也可以通过owo

子串看起来就很SuffixStructures

于是上SAM

本来想着直接LCT

后来发现没法串定位(暴力匹配复杂度不对)

然后就离线吧,先建出来然后链加子树和,树剖就odk。

其实更直接的套路是线段树合并right集合维护。

这个题很像BZOJ3881 那个题更复杂一点,需要树链的并(到时候看看这个题可不可以加强一下扔到校内模拟赛

当然还有正确的ACA写法,直接FAIL树上主席树。(就更像BZOJ3881了

(但感觉都很暴力,我的想法貌似还挺好的)

顺便学习了广义SAM的正确写法。(发现之前的写法死活改不对

//Love and Freedom.
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
#define inf 20021225
#define N 200010
#define pb push_back
using namespace std;
int read()
{
    int s=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
    return f*s;
}
struct node{int ch[26],fa,len;}t[N<<1];
struct edge{int to,lt;}e[N<<1];
int lt,poi,cnt,rt,ed[N],in[N<<1];
vector<int> endpos;
void insert(int c,int id)
{
    if(t[lt].ch[c])
    {
        int p=lt,np=lt=t[p].ch[c]; 
        if(t[np].len==t[p].len+1)    ed[id]=np,endpos.pb(np);
        else
        {
            int nq=++poi; ed[id]=nq; endpos.pb(nq); t[nq].len=t[p].len+1;
            memcpy(t[nq].ch,t[np].ch,sizeof(t[np].ch));
            t[nq].fa=t[np].fa; t[np].fa=nq;
            for(;p&&t[p].ch[c]==np;p=t[p].fa)    t[p].ch[c]=nq;
            lt=nq;
        }
        return;
    }
    int p=lt,np=lt=++poi; endpos.pb(np); ed[id]=np; t[np].len=t[p].len+1; 
    for(;p&&!t[p].ch[c];p=t[p].fa) t[p].ch[c]=np;
    if(!p){t[np].fa=rt; return;} int q=t[p].ch[c];
    if(t[q].len==t[p].len+1){t[np].fa=q; return;}
    int nq=++poi; t[nq].fa=t[q].fa; t[q].fa=t[np].fa=nq;
    memcpy(t[nq].ch,t[q].ch,sizeof(t[q].ch)); t[nq].len=t[p].len+1; 
    for(;p&&t[p].ch[c]==q;p=t[p].fa)    t[p].ch[c]=nq;
}
#define lowbit(x) (x&-x)
int tr[N<<1];
void add(int x,int v){while(x<=poi) tr[x]+=v,x+=lowbit(x);}
void modify(int l,int r,int v){add(l,v); add(r+1,-v);}
int query(int x){int a=0; while(x) a+=tr[x],x-=lowbit(x); return a;}
int dfn[N<<1],tot,sz[N<<1],son[N<<1],idfn[N<<1],top[N<<1],fa[N<<1];
void addedge(int x,int y){e[++cnt].to=y; e[cnt].lt=in[x]; in[x]=cnt;}
void build(){for(int i=2;i<=poi;i++)    addedge(t[i].fa,i);}
void dfs(int x,int fr)
{
    fa[x]=fr; sz[x]=1;
    for(int i=in[x];i;i=e[i].lt)
    {
        int y=e[i].to; dfs(y,x);
        sz[x]+=sz[y]; if(sz[y]>=sz[son[x]])    son[x]=y;
    }
}
void dfs2(int x,int t)
{
    top[x]=t; dfn[x]=++tot; idfn[tot]=x; if(!son[x])    return;    dfs2(son[x],t);
    for(int i=in[x];i;i=e[i].lt)    if(e[i].to!=son[x])    dfs2(e[i].to,e[i].to);
}
void modify(int x,int v)
{
    while(x)    modify(dfn[top[x]],dfn[x],v), x=fa[top[x]];
}
int ask(int x){return query(dfn[x]);}
#define Q 500010
int n,q; char ch[N]; int ans[Q],pre[N];
struct qry{int id,opt,pos,x;}r[Q<<1]; int qwq;
bool operator<(qry a,qry b){return a.pos<b.pos;}
int main()
{
    n=read(),q=read(); poi=rt=lt=1; int tot=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",ch+1); lt=rt; int sn=strlen(ch+1); 
        for(int j=1;j<=sn;j++)    ++tot,insert(ch[j]-'a',i);
        pre[i]=tot;
    }
    int L,R,K; build(); dfs(1,0); dfs2(1,1);
    for(int i=1;i<=q;i++)
        L=read(),R=read(),K=read(),
        r[++qwq].id=i,r[qwq].opt=-1,r[qwq].pos=pre[L-1],r[qwq].x=ed[K],
        r[++qwq].id=i,r[qwq].opt=1,r[qwq].pos=pre[R],r[qwq].x=ed[K];
    int p=1; sort(r+1,r+qwq+1);
    while(p<=qwq&&!r[p].pos)    p++;
    for(int i=0;i<endpos.size();i++)
    {
        int pos=endpos[i]; modify(pos,1);
        while(p<=qwq&&r[p].pos==i+1)    ans[r[p].id]+=r[p].opt*ask(r[p].x),p++;
    }
    for(int i=1;i<=q;i++)    printf("%d\n",ans[i]);
    return 0;
}
View Code

 

posted @ 2019-10-29 14:34  寒雨微凝  阅读(410)  评论(2编辑  收藏  举报