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