Codeforces 873F Forbidden Indices 字符串 SAM/(SA+单调栈)

原文链接https://www.cnblogs.com/zhouzhendong/p/9256033.html

题目传送门 - CF873F

题意

　　给定长度为 $n$ 的字符串 $s$，以及给定这个字符串每一个位置是否 “禁止结尾” 的信息。

　　一个字符串 $a$ 的价值为 $|a|\times f(a)$ 。

　　其中 $f(a)$为 $a$ 在 $s$ 中的匹配次数（如果匹配的结尾为禁止结尾点，那么不算匹配成功）

　　问在所有的字符串 $a$ 中，$\max(|a|\times f(a))$ 的值。

　　$n\leq 2\times 10 ^5$

题解

　　大概是好久没做 SAM 了，第一个想到的就是 SA 。其实用 SAM 做非常简单，不知道高到哪里去了……

　　这里讲 SA 做法。

　　首先考虑到 SA 对后缀开头点掌握的比较好，所以我们将原串翻转，把问题转化成开头禁止。

　　然后对 $s$ 串跑一下 SA 。

　　然后假装那些开头禁止的串不存在，就转化成一个经典的问题，直接单调栈搞定。

　　至于单调栈原理，这里有一道类似的运用单调栈的例题。

　　https://www.cnblogs.com/zhouzhendong/p/9026184.html

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=200005;
int n;
int SA[N],rank[N],tmp[N],height[N],tax[N];
int sum[N];
int st[N],top,pos[N];
char s[N],t[N];
void Sort(int n,int m){
    for (int i=0;i<=m;i++)
        tax[i]=0;
    for (int i=1;i<=n;i++)
        tax[rank[i]]++;
    for (int i=1;i<=m;i++)
        tax[i]+=tax[i-1];
    for (int i=n;i>=1;i--)
        SA[tax[rank[tmp[i]]]--]=tmp[i];
}
bool cmp(int rk[],int x,int y,int w){
    return rk[x]==rk[y]&&rk[x+w]==rk[y+w];
}
void Suffix_Array(char s[],int n){
    memset(SA,0,sizeof SA);
    memset(tmp,0,sizeof tmp);
    memset(rank,0,sizeof rank);
    memset(height,0,sizeof height);
    for (int i=1;i<=n;i++)
        rank[i]=s[i],tmp[i]=i;
    int m=234;
    Sort(n,m);
    for (int w=1,p=0;p<n;w<<=1,m=p){
        p=0;
        for (int i=n-w+1;i<=n;i++)
            tmp[++p]=i;
        for (int i=1;i<=n;i++)
            if (SA[i]>w)
                tmp[++p]=SA[i]-w;
        Sort(n,m);
        swap(rank,tmp);
        rank[SA[1]]=p=1;
        for (int i=2;i<=n;i++)
            rank[SA[i]]=cmp(tmp,SA[i],SA[i-1],w)?p:++p;
    }
    for (int i=1,j,k=0;i<=n;height[rank[i++]]=k)
        for (k=max(k-1,0),j=SA[rank[i]-1];s[i+k]==s[j+k];k++);
    height[1]=0;
}
int main(){
	scanf("%d",&n);
	scanf("%s%s",s+1,t+1);
	for (int i=1;i<=n/2;i++){
		swap(s[i],s[n-i+1]);
		swap(t[i],t[n-i+1]);
	}
	Suffix_Array(s,n);
	for (int i=1;i<=n;i++)
		if (t[SA[i]]=='1')
			sum[i]=sum[i-1];
		else
			sum[i]=sum[i-1]+1;
	LL ans=0;
	for (int i=1;i<=n;i++)
		if (t[i]=='0')
			ans=max(ans,(LL)n-i+1);
	top=1,st[top]=1,pos[top]=1;
	height[n+1]=0;
	for (int i=2;i<=n+1;i++){
		int j=st[top],p=i;
		while (height[j]>height[i]){
			ans=max(ans,1LL*height[j]*(sum[i-1]-sum[pos[top]-2]));
			p=pos[top];
			j=st[--top];
		}
		st[++top]=i,pos[top]=p;
	}
	printf("%I64d",ans);
	return 0;
}