●BZOJ 4566 [Haoi2016]找相同字符

题链:

http://www.lydsy.com/JudgeOnline/problem.php?id=4566

题解:

后缀数组,单调栈。
把两个串A,B拼接起来,中间用没出现过的字符隔开。
然后用倍增算法求出 sa[] rank[] height[]
接着用单调栈维护出两个数组 L[],R[],意义如下:
L[i]:表示在后缀数组中,排名最小(记其排名为 L[i])的后缀与排名为 i的后缀的LCP>=hei[i]
同理 R[i]:表示在后缀数组中,排名最大(记其排名为 R[i])的后缀与排名为 i的后缀的LCP>hei[i](注意这里没有‘=‘,避免重复)
显然对于 i后缀而言,其左边[L[i],i-1]的任意后缀与其右边(包括自己)[i,R[i]]的任意后缀的 LCP==hei[i]
(如果记 ANS[i]:表示为两个串的长度为 i的公共子串的个数(最后的答案就是 ANS[1]+ANS[2]+ANS[3]+....+ANS[N]))
那么 ANS[i]+=左边的A串后缀的数量*右边的B串后缀的数量 + 左边的B串后缀的数量*右边的A串后缀的数量
至于怎么求出左边和右边分别有多少个 A,B的子串,用一个前缀和(或者后缀和)就好了。
   
但是还没有完,
以上贡献的值都是建立在 LCP(最长公共前缀)上的,
但还有 LCP-1,LCP-2,LCP-3......这些没有地方被贡献到,
所以就要最后反向遍历一下答案数组,用后面的去更新前面的,得到的才是正确的 ANS[]:
ANS[i]+=ANS[i+1];
   
然后把 ANS[]加起来就是最终答案了。

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#define MAXN 405000
#define filein(x) freopen(#x".in","r",stdin);
#define fileout(x) freopen(#x".out","w",stdout);
using namespace std;
char S[MAXN],A[MAXN],B[MAXN];
int	sa[MAXN],rak[MAXN],hei[MAXN];
int L[MAXN],R[MAXN],sufA[MAXN],sufB[MAXN];
long long ANS[MAXN];
void build(int N,int M){
	static int cc[MAXN],ta[MAXN],tb[MAXN],*x,*y,h,p;
	x=ta; y=tb; h=0;
	for(int i=0;i<M;i++) cc[i]=0;
	for(int i=0;i<N;i++) cc[x[i]=S[i]]++;
	for(int i=1;i<M;i++) cc[i]+=cc[i-1];
	for(int i=N-1;i>=0;i--) sa[--cc[x[i]]]=i;
	for(int k=1;p=0,k<N;k<<=1){
		for(int i=N-k;i<N;i++) y[p++]=i;
		for(int i=0;i<N;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
		for(int i=0;i<M;i++) cc[i]=0;
		for(int i=0;i<N;i++) cc[x[y[i]]]++;
		for(int i=1;i<M;i++) cc[i]+=cc[i-1];
		for(int i=N-1;i>=0;i--) sa[--cc[x[y[i]]]]=y[i];
		swap(x,y); y[N]=-1; x[sa[0]]=0; M=1;
		for(int i=1;i<N;i++)
			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?M-1:M++;
		if(M>=N) break;
	}
	for(int i=0;i<N;i++) rak[sa[i]]=i/*,puts(S+sa[i])*/;
	for(int i=0,j;i<N;i++){
		if(h) h--;
		if(rak[i]){
			j=sa[rak[i]-1];
			while(S[i+h]==S[j+h]) h++;
		}
		hei[rak[i]]=h;
	}
}
void preLR(int N){
	static int stk[MAXN],stp[MAXN],top;
	stp[top=0]=0;
	for(int i=0;i<N;i++){
		while(top&&stk[top]>=hei[i]) top--;
		L[i]=stp[top]; top++;
		stk[top]=hei[i]; stp[top]=i;
	}
	stp[top=N]=N;
	for(int i=N-1;i>=0;i--){
		while(top<N&&stk[top]>hei[i]) top++;
		R[i]=stp[top]-1; top--;
		stk[top]=hei[i]; stp[top]=i;
	}
}
int main()
{
	int N=0,lA,lB; long long ans=0;
	scanf("%s",A); lA=strlen(A);
	scanf("%s",B); lB=strlen(B);
	for(int i=0;i<lA;i++) S[N+i]=A[i]; S[lA]=‘#‘; N+=lA+1;
	for(int i=0;i<lB;i++) S[N+i]=B[i]; N+=lB;
	//puts(S); printf("%d",N);
	build(N,300);
	for(int i=N-1;i>=0;i--) sufA[i]=sufA[i+1]+(sa[i]<lA);
	for(int i=N-1;i>=0;i--) sufB[i]=sufB[i+1]+(sa[i]>lA);
	preLR(N);
	for(int i=0,l,r;i<N;i++){
		l=L[i]; r=R[i];
		ANS[hei[i]]+=1ll*(sufA[l]-sufA[i])*(sufB[i]-sufB[r+1]);
		ANS[hei[i]]+=1ll*(sufB[l]-sufB[i])*(sufA[i]-sufA[r+1]);
	}
	for(int i=N-1;i>=1;i--){
		ANS[i]+=ANS[i+1];
		ans+=ANS[i];
	}
	printf("%lld",ans);
	return 0;
}
时间: 2024-10-02 01:32:37

●BZOJ 4566 [Haoi2016]找相同字符的相关文章

bzoj 4566: [Haoi2016]找相同字符

Description 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两 个子串中有一个位置不同. Input 两行,两个字符串s1,s2,长度分别为n1,n2.1 <=n1, n2<= 200000,字符串中只有小写字母 Output 输出一个整数表示答案 Sample Input aabb bbaa Sample Output 10 HINT Source 首先这种多个字符串的一般要拼成一个串: 如果考虑暴力的话,那就是开头在第一个串的后

4566: [Haoi2016]找相同字符 SAM

折腾了好久.不过收获还是很多的.第一次自己去画SAM所建出来fail树.深入体会了这棵树的神奇性质. 当然,我最终靠着自己A掉了.(这是我第一次推SAM的性质(以前都是抄别人的,感觉自己好可耻),不过感觉好像是摸着黑行走啊!) 这道题,可以先对第一个串建出后缀自动机.然后第二个串在后缀自动机上跑. 首先,SAM所建出的fail树的性质有: 1: 树上一个节点对应了多个串,串的个数是 len[x] - len[fa[x]], 同时它们都出现了 sz[x] 次(感觉好像说不大清,可以看一下代码对于s

BZOJ4566:[Haoi2016]找相同字符

4566: [Haoi2016]找相同字符 Time Limit: 20 Sec  Memory Limit: 256 MBSubmit: 545  Solved: 302[Submit][Status][Discuss] Description 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两 个子串中有一个位置不同. Input 两行,两个字符串s1,s2,长度分别为n1,n2.1 <=n1, n2<= 200000,字符串中只有小写字母

P3181 [HAOI2016]找相同字符

P3181 [HAOI2016]找相同字符 对一个串建SAM,另一个串在这上面跑,到达一点时,假设经过了\(cnt\)个点 计算这个串所有后缀产生的贡献就好了,直接暴力跑上去可能会超时,topsort预处理一下 #include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<algorithm> #incl

[HAOI2016]找相同字符(后缀数组+单调栈)

[HAOI2016]找相同字符(后缀数组+单调栈) 题面 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两个子串中有一个位置不同. 分析 我们把两个字符串接在一起,中间加一个分隔符.如\(\text{AABB}\)和\(\text{BBAA}\)变成\(\text{AABB|BBAA}\).我们考虑两个相同字串,如\(\text{BB}\),它在新串中对应了两个后缀\(BB|BBAA\)和\(\text{BBAA}\)的LCP. 容易发现,LC

[HAOI2016]找相同字符(广义SAM)

[HAOI2016]找相同字符(广义SAM) 题面 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两个子串中有一个位置不同. 分析 此题有一个比较繁琐的后缀数组做法,但是用广义SAM可以秒杀. 把两个串建成广义SAM,对于每个后缀,记录\(endpos\)集合中落在第一个串中和第二个串中的位置个数,记为\(cnt_{x,0},cnt_{x,1}\). 对于自动机上的每个节点\(x\),出现位置方案数的贡献是\(cnt_{x,0} \cdot c

BZOJ4566:[HAOI2016]找相同字符——题解

https://www.lydsy.com/JudgeOnline/problem.php?id=4566 https://www.luogu.org/problemnew/show/P3181 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两个子串中有一个位置不同. 广义后缀自动机,两个串各处理他们的size(或right?),然后对结点l排序,对于每个结点他们的size相乘即为答案. ……等等怎么WA了啊. 比如: aba abaa 这组数

[HAOI2016]找相同字符 广义后缀自动机_统计出现次数

题目描述:给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两个子串中有一个位置不同. 输入输出格式输入格式:两行,两个字符串 s1,s2,长度分别为n1,n2.1 <=n1, n2<= 200000,字符串中只有小写字母 输出格式:输出一个整数表示答案 题解:对 $2$ 个字符串建立一个广义后缀自动机.实际上,广义后缀自动机就是对多个字符串用一个自动机加以维护.每加入完毕一个字符串时,将 $last$ 设为 $1$.插入字符时,若 $ch[la

[HAOI2016]找相同字符(SAM)

题目描述 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两个子串中有一个位置不同. 输入输出格式 输入格式: 两行,两个字符串s1,s2,长度分别为n1,n2.1 <=n1, n2<= 200000,字符串中只有小写字母 输出格式: 输出一个整数表示答案 输入输出样例 输入样例#1: 复制 aabb bbaa 输出样例#1: 复制 10 这到题目是vj上一道题目的简化版对第一个串建立自动机 在拓扑一边球每个状态串的出现次数然后然第二个串在树上