height led cor += cmp const cstring 每一個 !=

Description

給定兩個字符串，求出在兩個字符串中各取出一個子串使得這兩個子串相同的方案數。兩個方案不同當且僅當這兩 個子串中有一個位置不同。

Input

兩行，兩個字符串s1，s2，長度分別為n1，n2。1 <=n1, n2<= 200000，字符串中只有小寫字母

Output

輸出一個整數表示答案

Sample Input

aabb
bbaa

Sample Output

10

HINT

Source

首先這種多個字符串的一般要拼成一個串； 如果考慮暴力的話，那就是開頭在第一個串的後綴和開頭在第二個串的後綴的LCP和； 可以通過ST表做到n^2; 考慮枚舉子串的長度L,我們需要知道對於每一個L,滿足開頭在第一個串的後綴和開頭在第二個串的後綴的LCP等於L的個數； 對於確定了height的值的求寫鬼東西時候，可以考慮品酒大會的套路，按height建立並查集然後從大到小合並即可； 這個題的話，只需要維護每個並查集中開頭在第一個串中的數量和開頭在第二個串中的數量，合並的時候乘起來即可；

// MADE BY QT666
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=400050;
int gi(){
  int x=0,flag=1;
  char ch=getchar();
  while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘) flag=-1;ch=getchar();}
  while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();
  return x*flag;
}
int sa[N],len,y[N],rk,rnk[N],height[N],pre[N],pre2[N],ST[N][20],cor[N],fa[N];
char ch[N],ch2[N],a[N];
ll ans[N],num[N][3],Ans;
struct data{
  int fir,sec,id;
}x[N];
bool cmp(const data &a,const data &b){
  if(a.fir==b.fir) return a.sec<b.sec;
  else return a.fir<b.fir;
}
void work2(){
  rk=1;y[x[1].id]=rk;
  for(int i=2;i<=len;i++){
    if(x[i-1].fir!=x[i].fir||x[i-1].sec!=x[i].sec) rk++;
    y[x[i].id]=rk;
  }
}
void work(){
  sort(x+1,x+1+len,cmp);work2();
  for(int i=1;i<=len;i<<=1){
    for(int j=1;j+i<=len;j++) x[j].fir=y[j],x[j].sec=y[j+i],x[j].id=j;
    for(int j=len-i+1;j<=len;j++) x[j].fir=y[j],x[j].sec=0,x[j].id=j;
    sort(x+1,x+1+len,cmp);work2();
    if(rk==len) break;
  }
}
void get_height(){
  int kk=0;for(int i=1;i<=len;i++) rnk[sa[i]]=i;
  for(int i=1;i<=len;i++){
    if(kk) kk--;
    int j=sa[rnk[i]-1];
    while(a[i+kk]==a[j+kk]) kk++;
    height[rnk[i]]=kk;
  }
}
struct Data{
  int l,r,hi;
}g[N];
bool cmp2(const Data &a,const Data &b){
  return a.hi>b.hi;
}
int find(int x) {
  if(x!=fa[x]) fa[x]=find(fa[x]);
  return fa[x];
}
void merge(int x,int y,int hi){
  ans[hi]+=num[x][1]*num[y][2]+num[x][2]*num[y][1];
  fa[x]=y;num[y][1]+=num[x][1],num[y][2]+=num[x][2];
}
int main(){
  scanf("%s",ch+1);scanf("%s",ch2+1);int len1=strlen(ch+1),len2=strlen(ch2+1);
  len=len1+len2+1;for(int i=1;i<=len1;i++) a[i]=ch[i],cor[i]=1;
  a[len1+1]=‘#‘;for(int i=1;i<=len2;i++) a[i+len1+1]=ch2[i],cor[i+len1+1]=2;
  for(int i=1;i<=len;i++) x[i].id=i,x[i].fir=x[i].sec=a[i]-‘a‘+1;
  work();for(int i=1;i<=len;i++) sa[y[i]]=i;
  get_height();for(int i=1;i<=len;i++) fa[i]=i,num[i][cor[i]]++;
  for(int i=2;i<=len;i++) g[i-1]=(Data){sa[i-1],sa[i],height[i]};
  sort(g+1,g+len,cmp2);
  for(int i=1;i<len;i++){
    int x=find(g[i].l),y=find(g[i].r);
    if(x!=y) merge(x,y,g[i].hi);
  }
  for(int i=1;i<=len;i++) Ans+=ans[i]*i;
  printf("%lld\n",Ans);
  return 0;
}

　　

bzoj 4566: [Haoi2016]找相同字符