CF873F Forbidden Indices 后缀自动机+水题

最新推荐文章于 2024-07-07 21:22:23 发布

原创最新推荐文章于 2024-07-07 21:22:23 发布 · 134 阅读

0 ·

CC 4.0 BY-SA版权

本文深入探讨了AC自动机的实现细节，通过一个具体的代码示例，详细解释了如何使用AC自动机处理字符串匹配问题。文章涵盖了AC自动机的构建过程、节点结构以及关键的extend函数，展示了其在高效字符串搜索中的应用。

刷刷水~

Code:

#include <cstdio> 
#include <cstring> 
#include <algorithm>
#define N 200005
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;  
char str1[N],str2[N]; 
int tot,last,n,c[N<<1],rk[N<<1]; 
struct Node 
{
    int ch[26],f,len,cnt; 
}t[N<<1];    
void extend(int c,int flag) 
{
    int np=++tot,p=last; 
    last=np,t[np].len=t[p].len+1;     
    for(;p&&!t[p].ch[c];p=t[p].f) t[p].ch[c]=np;    
    if(!p) t[np].f=1; 
    else
    {
        int q=t[p].ch[c]; 
        if(t[q].len==t[p].len+1) t[np].f=q; 
        else 
        {
            int nq=++tot; 
            t[nq].len=t[p].len+1; 
            t[nq].f=t[q].f,t[q].f=t[np].f=nq;    
            memcpy(t[nq].ch,t[q].ch,sizeof(t[q].ch));   
            for(;p&&t[p].ch[c]==q;p=t[p].f) t[p].ch[c]=nq;    
        }
    }
    t[np].cnt=flag;  
}
int main() 
{ 
    int i,j; 
    long long answer=0; 
    // setIO("input");  
    scanf("%d%s%s",&n,str1+1,str2+1); 
    for(last=tot=i=1;i<=n;++i) extend(str1[i]-'a',str2[i]=='0'?1:0);    
    for(i=1;i<=tot;++i) ++c[t[i].len]; 
    for(i=1;i<=tot;++i) c[i]+=c[i-1];
    for(i=1;i<=tot;++i) rk[c[t[i].len]--]=i;    
    for(i=tot;i>=1;--i) 
    {  
        int p,ff; 
        p=rk[i],ff=t[p].f;   
        answer=max(answer,(long long)t[p].len*t[p].cnt);    
        t[ff].cnt+=t[p].cnt;   
    }
    printf("%lld\n",answer);    
    return 0; 
}