Codeforces F. Forbidden Indices（后缀自动机）

题目链接

题解：

建立后缀自动机，根据第二个字符串的01情况对后缀自动机上的点进行标记，排序后让出现次数和标记一起传给它的fa，最后统计时去掉标记即可。

---

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN = 4e5+50;
int nxt[MAXN][26],fa[MAXN],len[MAXN],cnt[MAXN];
int last=1,tot=1;
char s[MAXN],mark[MAXN];
int tag[MAXN],a[MAXN],c[MAXN];
inline void Insert(int x,int k){
    int p=last,np=++tot;
    last=np,len[np]=len[p]+1;
    for(;p&&!nxt[p][x];p=fa[p]) nxt[p][x]=np;
    if(!p) fa[np]=1;
    else{
        int q=nxt[p][x];
        if(len[p]+1==len[q]) fa[np]=q;
        else{
            int nq=++tot;
            len[nq]=len[p]+1;
            memcpy(nxt[nq],nxt[q],sizeof(nxt[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for(;nxt[p][x]==q;p=fa[p]) nxt[p][x]=nq;
        }
    }
    cnt[np]=1,tag[np]=k;
}
inline void Sort(){
    for(int i=1;i<=tot;i++) c[len[i]]++;
    for(int i=1;i<=tot;i++) c[i]+=c[i-1];
    for(int i=tot;i;i--) a[c[len[i]]--]=i;
    for(int i=tot;i;i--) cnt[fa[a[i]]]+=cnt[a[i]],tag[fa[a[i]]]+=tag[a[i]];
}
int main(){
    int n; scanf("%d",&n);
    scanf("%s%s",s,mark);
    for(int i=0;i<n;i++) Insert(s[i]-'a',mark[i]-'0');
    Sort();
    LL res = 0;
    for(int i=1;i<=tot;i++)
        res = max(res,1LL*(cnt[i]-tag[i])*len[i]);
    printf("%lld\n",res);
    return 0;
}