BZOJ 4566 JZYZOJ 1547 [haoi2016T5]找相同子串 后缀数组 并查集

http://172.20.6.3/Problem_Show.asp?id=1547

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

单纯后缀数组是O(n^2)应该是40分

似乎后缀自动机是正解。

但是后缀数组+并查集也可以乱搞a掉，这里写的是并查集写法，也算是get了一个并查集的用法，某种意义上并查集可以用来维护区间最大值最小值的贡献，实现方法见代码。

定义字符串大小的整型变量时候，

char siz;

导致re什么的，

我大概是个zz。

顺便存个板子，抄紫萱学姐的板子。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 const int maxn=;
 const int pl=;
 int sa[maxn*+pl]={};//排名第i的是从sa[i]开始的数组
 int rk[maxn*+pl]={};//i的排名
 int height[maxn*+pl]={};//排名第i的与排名第i-1的最长相同前缀长度
 int temp[maxn*+pl]={};//暂时的排名
 int cnt[maxn*+pl]={};//第i种（字典序）前缀的有多少个（的前缀和）
 int p[maxn*+pl]={};//临时对这一次需要用的sa的储存，处理了后缀长度不同的情况。
 char ch[maxn*+pl]={},ch1[maxn]={},ch2[maxn]={};
 int siz1,siz2,siz;
 int fa[maxn*+pl]={},a[maxn*+pl]={},lef[maxn*+pl]={},rig[maxn*+pl]={};
 inline bool equ(int x,int y,int l){return rk[x]==rk[y]&&rk[x+l]==rk[y+l];}
 void SA(){
     for(int i=;i<=siz;i++)rk[i]=ch[i],sa[i]=i;
     for(int i,sig=,l=,pos=;pos<siz;sig=pos){//l从0开始是预处理
         pos=;
         for(i=siz-l+;i<=siz;i++)p[++pos]=i;
         for(i=;i<=siz;i++)if(sa[i]>l)p[++pos]=sa[i]-l;
         for(i=;i<=sig;i++)cnt[i]=;
         for(i=;i<=siz;i++)cnt[rk[p[i]]]++;
         for(i=;i<=sig;i++)cnt[i]+=cnt[i-];
         for(i=siz;i>;i--){sa[cnt[rk[p[i]]]]=p[i];cnt[rk[p[i]]]--;}
         pos=;
         for(i=;i<=siz;i++){
             if(equ(sa[i],sa[i-],l))temp[sa[i]]=pos ;
             else temp[sa[i]]=++pos;
         }for(i=;i<=siz;i++)rk[i]=temp[i];
         if(l==)l=;
         else l<<=;
     }
     for(int i=,k=;i<=siz;i++){
         /*对于每一个位置的后缀，下一个位置的后缀可匹配的最短长度
         一定大于等于该位置可匹配的长度-1,显然。所以是O(n)的算法
         */
         if(rk[i]==){k=;continue;}
         if(k>)k--;
         int j=sa[rk[i]-];
         while(ch[i+k]==ch[j+k])k++;
         height[rk[i]]=k;
     }
 }
 bool mcmp(int x,int y){return height[x]>height[y];}
 int getfa(int x){
     if(x!=fa[x])fa[x]=getfa(fa[x]);
     return fa[x];
 }
 int main(){
     //freopen("a.in","r",stdin);
     scanf("%s",&ch1);siz1=strlen(ch1);
     scanf("%s",&ch2);siz2=strlen(ch2);
     ch[siz1+]='z'+;siz=siz1+siz2+;
     for(int i=;i<siz1;i++)ch[i+]=ch1[i];
     for(int i=;i<siz2;i++)ch[siz1++i]=ch2[i];
     SA();
     for(int i=;i<=siz;i++){
         a[i]=fa[i]=i;
         lef[i]=(sa[i]<=siz1);
         rig[i]=^lef[i];
     }sort(a+,a++siz,mcmp);
     long long ans=;
     int x,y;
     for(int i=;i<=siz;i++){
         if(a[i]==)continue;
         x=getfa(a[i]);y=getfa(a[i]-);
         ans+=(long long)height[a[i]]*(long long)(lef[x]*rig[y]+rig[x]*lef[y]);
         lef[x]+=lef[y];rig[x]+=rig[y];fa[y]=x;
     }printf("%lld\n",ans);
     return ;
 }

更新:http://www.cnblogs.com/137shoebills/p/8511439.html 这是一道板子题的代码，注释应该被我完善了，更加清晰一点，所以我为什么要先写一道组合题再写板子啊喂。