【洛谷P4081】[USACO17DEC]Standing Out from the Herd(广义SAM)

传送门


题解:

比较套路的题,建广义SAM然后暴跳fail更新所属串集合。

所属串集合大小为1的保留那个元素,否则存为-1。最后直接把答案加到对应串上即可。


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

cs int N=1e5+7;

int n;
char s[N];int len;
int L[N],ans[N];

namespace SAM{
	cs int N=::N<<1|1;
	int fa[N],len[N],son[N][26],vis[N],now=1;
	inline int work(int p,int c){
		int nq=++now,q=son[p][c];
		memcpy(son[nq],son[q],sizeof son[q]);
		len[nq]=len[p]+1,fa[nq]=fa[q],fa[q]=nq;
		for(;p&&son[p][c]==q;p=fa[p])son[p][c]=nq;
		return nq;
	}
	inline int push_back(int p,int c){
		if(son[p][c]){
			if(len[son[p][c]]==len[p]+1)return son[p][c];
			else return work(p,c);
		}
		int cur=++now;len[cur]=len[p]+1;
		for(;p&&!son[p][c];p=fa[p])son[p][c]=cur;
		if(!p)fa[cur]=1;
		else if(len[son[p][c]]==len[p]+1)fa[cur]=son[p][c];
		else fa[cur]=work(p,c);
		return cur;
	}
	inline void update(int u,int id){
		for(;u&&vis[u]!=id&&vis[u]!=-1;u=fa[u])vis[u]=vis[u]?-1:id;
	}
	inline void solve(){
		for(int re i=1;i<=n;++i)
		for(int re cur=1,j=L[i-1]+1;j<=L[i];++j)
		update(cur=son[cur][s[j]-'a'],i);
		for(int re i=1;i<=now;++i)if(vis[i]>0)ans[vis[i]]+=len[i]-len[fa[i]];
	}
}

signed main(){
#ifdef zxyoi
	freopen("herd.in","r",stdin);
#endif
	scanf("%d",&n);
	for(int re i=1;i<=n;++i){
		scanf("%s",s+L[i-1]+1);
		L[i]=strlen(s+L[i-1]+1)+L[i-1];int p=1;
		for(int re j=L[i-1]+1;j<=L[i];++j)
		p=SAM::push_back(p,s[j]-'a');
	}
	SAM::solve();
	for(int re i=1;i<=n;++i)std::cout<<ans[i]<<"\n";
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值