CF873F Forbidden Indices 后缀自动机+水题

刷刷水~

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;
}