bzoj4516: [Sdoi2016]生成魔咒

SAM算法详解

最新推荐文章于 2020-01-13 20:23:43 发布

转载最新推荐文章于 2020-01-13 20:23:43 发布 · 49 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/AKCqhzdy/p/8693601.html

本文详细介绍了字符串匹配自动机(SAM)的实现原理与代码细节。通过使用map存储字符转移信息，该算法能够在处理输入字符串时动态构建状态机，并计算每个状态对应的答案累加值。适用于文本处理、模式匹配等场景。

就是SAM嘛。

答案累加就是dep[parent[now]]-dep[now]

有个比较坑的就是要用map。。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<map>
using namespace std;
typedef long long LL;

int a[210000];
map<int,int>ch[410000];int root,last,cnt;
LL ans;
int parent[410000],dep[410000];
void add(int k)
{
    int x=a[k];
    int now=++cnt,p=last;
    dep[now]=k;
    
    while(p!=0&&ch[p][x]==0)
        ch[p][x]=now,p=parent[p];
        
    if(p==0)parent[now]=root;
    else 
    {
        int pre=ch[p][x];
        if(dep[pre]==dep[p]+1)parent[now]=pre;
        else
        {
            int npre=++cnt;
            ch[npre]=ch[pre];
            parent[npre]=parent[pre];
            dep[npre]=dep[p]+1;
            
            parent[now]=parent[pre]=npre;
            while(p!=0&&ch[p][x]==pre)
                ch[p][x]=npre,p=parent[p];
        }
    }
    last=now;
    if(parent[now]==0)ans+=(LL(dep[now]));
    else ans+=(LL(dep[now]-dep[parent[now]]));
}
int main()
{
    int n;
    scanf("%d",&n);
    root=last=cnt=1;ans=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]),add(i);
        printf("%lld\n",ans);
    }
    return 0;
}