HDU-4416 Good Article Good sentence (後綴自動機)

HDU-4416 Good Article Good sentence (後綴自動機)

題意：給定一個串\(A\)和若干串\(B_i\)，求出如今\(A\)中且不出如今\(B\)中的子串個數

這個題咱們不能把\(B_i\)接在一塊兒作，會直接MLE!

對於\(A\)構建\(SAM\)

考慮對於每一個\(B_i\)把出如今\(B_i\)中的子串踢掉

就是一個匹配問題，對於\(B_i\)的每個前綴，在\(A\)中匹配一個最長的後綴，更新它的\(max\)值，即最長的被\(B_i\)匹配的長度

這樣的更新顯然是須要對於一個狀態到\(parent/link\)樹的根的路徑都進行一遍，因此能夠最後子樹累\(max\)

那麼每一個狀態剩下的\(max+1-len\)這一部分就是合法的

Tip:這個題用讀優好像會被卡掉

#include<bits/stdc++.h>
using namespace std;

#define reg register
typedef long long ll;
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(int i=a,i##end=b;i>=i##end;--i)

#define pb push_back
template <class T> inline void cmin(T &a,T b){ ((a>b)&&(a=b)); }
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); }

const int N=2e5+10;

bool be;

int n;
char s[N];
int trans[N][26];
int lst,stcnt;
int len[N],link[N],ma[N];

struct Edge{
    int to,nxt;
}e[N];
int head[N],ecnt;
void AddEdge(int u,int v) {
    ecnt++;
    e[ecnt].to=v,e[ecnt].nxt=head[u];
    head[u]=ecnt;
}

ll ans;
int vis[N];

bool ed;

void Init(){ 
    link[0]=-1,len[0]=0;
    rep(i,0,stcnt) {
        len[i]=head[i]=ma[i]=0;
        rep(j,0,25) trans[i][j]=0;
        vis[i]=0;
    }
    stcnt=lst=ecnt=0;
}

void Extend(int c) {
    int cur=++stcnt,p=lst;
    len[cur]=len[lst]+1;
    while(~p && !trans[p][c]) trans[p][c]=cur,p=link[p];
    if(p==-1) link[cur]=0;
    else {
        int q=trans[p][c];
        if(len[q]==len[p]+1) link[cur]=q;
        else {
            int clone=++stcnt;
            link[clone]=link[q];
            memcpy(trans[clone],trans[q],104);
            len[clone]=len[p]+1;
            while(~p && trans[p][c]==q) trans[p][c]=clone,p=link[p];
            link[q]=link[cur]=clone;
        }
    }
    lst=cur;
}

void dfs(int u) {
    if(vis[u]) return;
    vis[u]=1;
    for(reg int i=head[u];i;i=e[i].nxt) {
        int v=e[i].to;
        dfs(v);
        cmax(ma[u],ma[v]);
    }
    ans+=max(0,len[u]-max(len[link[u]],ma[u]));
}

int main(){
    int T;cin>>T;
    rep(qaq,1,T) {
        cin>>n;
        scanf("%s",s+1);
        int len=strlen(s+1);
        Init();
        rep(i,1,len) Extend(s[i]-'a');
        rep(qwq,1,n) {
            scanf("%s",s+1);
            len=strlen(s+1);
            int p=0,nowlen=0;
            rep(i,1,len) {
                int c=s[i]-'a';
                while(p && !trans[p][c]) p=link[p];
                cmin(nowlen,::len[p]);
                p=trans[p][c];
                nowlen++;// 匹配而且更新
                cmax(ma[p],nowlen);
            }
        }
        rep(i,1,stcnt) AddEdge(link[i],i);
        ans=0;
        dfs(0);
        printf("Case %d: %lld\n",qaq,ans);
    }
}