【SAM】BZOJ5137 [Usaco2017 Dec] Standing Out from the Herd

【题目】
lydsy
给定 n n n个字符串,对于每个字符串,问只在这个字符串中出现的子串有多少个。
n , ∑ ∣ S ∣ ≤ 1 0 5 n,\sum|S|\leq 10^5 n,S105

【解题思路】
随便写写板子吧。
建出广义 SAM \text{SAM} SAM后自下而上合并 right \text{right} right集合即可,实现时就是给每个结束节点打个颜色标记,若标记相同就设成 − 1 -1 1就好了。

试图冲榜一但失败了。

【参考代码】

#include<bits/stdc++.h>
using namespace std;

const int N=2e5+10;
int n,f[N],b[N],c[N];
long long ans[N];
char s[N];

struct SAM
{
	int sz,las,fa[N],mx[N],ch[N][26];
	void extend(int x)
	{
		int p,np,q,nq;
		if(ch[las][x])
		{
			p=las;las=q=ch[p][x];//be careful here, las=ch[p][x]
			if(mx[q]==mx[p]+1) return;
			las=nq=++sz;mx[nq]=mx[p]+1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			fa[nq]=fa[q];fa[q]=nq;
			for(;p && ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
			return;
		}
		p=las;np=las=++sz;mx[np]=mx[p]+1;
		for(;p && !ch[p][x];p=fa[p]) ch[p][x]=np;
		if(!p) fa[np]=1;
		else
		{
			q=ch[p][x];
			if(mx[q]==mx[p]+1) fa[np]=q;
			else
			{
				nq=++sz;mx[nq]=mx[p]+1;
				memcpy(ch[nq],ch[q],sizeof(ch[q]));
				fa[nq]=fa[q];fa[q]=fa[np]=nq;
				for(;p && ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
			}
		}
	}
	void merge()
	{
		for(int i=1;i<=sz;++i) b[mx[i]]++;
		for(int i=1;i<=sz;++i) b[i]+=b[i-1];
		for(int i=sz;i;--i) c[b[mx[i]]--]=i;
		for(int i=sz,x;i;--i)
		{
			x=c[i];
			if(~f[x]) ans[f[x]]+=mx[x]-mx[fa[x]];
			if(f[fa[x]] && f[fa[x]]!=f[x]) f[fa[x]]=-1; else f[fa[x]]=f[x];
		} 
	}
}T;

int main()
{
#ifdef Durant_Lee
	freopen("BZOJ5137.in","r",stdin);
	freopen("BZOJ5137.out","w",stdout);
#endif
	scanf("%d",&n);T.sz=1;
	for(int i=1;i<=n;++i)
	{
		T.las=1;scanf("%s",s+1);int len=strlen(s+1);
		for(int j=1;j<=len;++j) 
		{
			T.extend(s[j]-'a');
			if(!f[T.las]) f[T.las]=i; else f[T.las]=-1;
		}
	}
	T.merge();
	for(int i=1;i<=n;++i) printf("%lld",ans[i]),putchar('\n');
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值