codeforces 666E. Forensic Examination

题意:

给一些模式串,每次询问询问串的子串在 [l,r] [ l , r ] 中的哪个模式串出现最多。

题解:

好久久久久久久久久久久久久没写字符串相关,调的简直心态爆炸。
直接建出广义sam,每次就相当于询问一个子树的众数。
用可持久化线段树合并即可。
具体细节见代码,线段树合并很久没写写错几次。
code:

#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
const int inf=1<<28;
struct SAM{
    int a[26],par,max;
}sam[100010];int tot=1,Root=1,tail=1,fa[100010][22],dep[100010];
int n,m;
struct Node{
    int y,next;
}a[100010];int len=0,last[100010];
struct node{
    int x,c;
    node() {x=inf;c=0;}
    node(int a,int b) {x=a;c=b;}
};
bool operator > (node a,node b) {return a.c==b.c?a.x<b.x:a.c>b.c;}
node mymax(node a,node b) {return a>b?a:b;}
struct trnode{
    int lc,rc;
    node c;
}tr[6000010];int trtot=0,root[100010];
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
void update(int &x,int l,int r,int k)
{
    if(!x) x=++trtot,tr[x].c.x=l;
    if(l==r) {tr[x].c.c++;return;}
    int mid=(l+r)/2;
    if(k<=mid) update(tr[x].lc,l,mid,k);
    else update(tr[x].rc,mid+1,r,k);
    tr[x].c=mymax(tr[tr[x].lc].c,tr[tr[x].rc].c);
}
int addsam(int c,int p,int k)
{
    int np=++tot;sam[np].max=sam[p].max+1;
    for(;p&&!sam[p].a[c];p=sam[p].par) sam[p].a[c]=np;
    tail=np;update(root[np],1,m,k);
    if(!p) sam[np].par=Root;
    else
    {
        int q=sam[p].a[c];
        if(sam[q].max==sam[p].max+1) sam[np].par=q;
        else
        {
            int nq=++tot;
            sam[nq]=sam[q];sam[nq].max=sam[p].max+1;
            sam[np].par=sam[q].par=nq;
            for(;p&&sam[p].a[c]==q;p=sam[p].par) sam[p].a[c]=nq;
        }
    }
    return np;
}
int merge(int x1,int x2,int l,int r)
{
    if(!x1||!x2) {return x1+x2;}
    int x=++trtot;tr[x]=tr[x1];
    if(l==r) tr[x].c.c=tr[x1].c.c+tr[x2].c.c;
    else
    {
        int mid=(l+r)/2;
        tr[x].lc=merge(tr[x1].lc,tr[x2].lc,l,mid);
        tr[x].rc=merge(tr[x1].rc,tr[x2].rc,mid+1,r);
        tr[x].c=mymax(tr[tr[x].lc].c,tr[tr[x].rc].c);
    }
    return x;
}
char s[500010],s1[50010];
int pre[500010],f[500010];
node findans(int x,int l,int r,int fl,int fr)
{
    if(!x) return node(fl,0);
    if(fl==l&&fr==r) return tr[x].c;
    int mid=(l+r)/2;
    if(fr<=mid) return findans(tr[x].lc,l,mid,fl,fr);
    if(fl>mid) return findans(tr[x].rc,mid+1,r,fl,fr);
    return mymax(findans(tr[x].lc,l,mid,fl,mid),findans(tr[x].rc,mid+1,r,mid+1,fr));
}
void dfs(int x)
{
    fa[x][0]=sam[x].par;dep[x]=dep[fa[x][0]]+1;
    for(int i=1;(1<<i)<=dep[x];i++)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        dfs(y);
        root[x]=merge(root[x],root[y],1,m);
    }
}
void solve(int l,int r,int fl,int fr)
{
    int x=pre[fr],len=fr-fl+1;
    if(f[fr]<len) {printf("%d 0\n",l);return;}
    for(int i=20;i>=0;i--)
        if((1<<i)<=dep[x]&&sam[fa[x][i]].max>=len) x=fa[x][i];
    node t=findans(root[x],1,m,l,r);
    printf("%d %d\n",t.x,t.c);
}
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    scanf("%d",&m);getchar();
    for(int i=1;i<=m;i++)
    {
        scanf("%s",s1+1);
        int Len=strlen(s1+1);int p=Root;
        for(int j=1;j<=Len;j++) p=addsam(s1[j]-'a',p,i);
    }
    int x=Root,now=0;
    for(int i=1;i<=n;i++)
    {
        int c=s[i]-'a';
        while(x!=Root&&!sam[x].a[c]) x=sam[x].par,now=sam[x].max;
        if(sam[x].a[c]) x=sam[x].a[c],now++;
        pre[i]=x;f[i]=now;
    }
    for(int i=2;i<=tot;i++) ins(sam[i].par,i);
    dep[0]=-1;dfs(Root);
    int Q;scanf("%d",&Q);
    while(Q--)
    {
        int l,r,fl,fr;scanf("%d %d %d %d",&l,&r,&fl,&fr);
        solve(l,r,fl,fr);
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值