2601 调皮的小Biu(后缀数组,倍增算法,计算最长公共字串,板子)

小Biu的期中考试刚刚结束,调皮的小Biu不喜欢学习,所以他考试中抄袭了小Piu的试卷。
考试过程中一共有n道题目,每道题目的标准答案为区间[1,5]中的一个正整数。
现在有小Piu和小Biu的答案序列a和b,现在老师想知道两个答案序列最长相等的连续子串的长度是多少。
比如一共有10道题,a序列为(1 1 2 1 2 1 2 1 1 5),b序列为(3 3 2 3 1 2 1 1 3 4),则最长相等的子串为(1,2,1,1),所以答案为4。
输入
第1行:一个正整数n,表示题目的个数。(1<=n<=100000)
第2行:n个正整数,第i个正整数表示a[i]。(1<=a[i]<=5)
第3行:n个正整数,第i个正整数表示b[i]。(1<=b[i]<=5)
输出
输出一个正整数表示两个序列的最长相等子串的长度。
输入样例

10
1 1 2 1 2 1 2 1 1 5
3 3 2 3 1 2 1 1 3 4
输出样例
4
用普通的dp会超时,只能用后缀数组,具体的原理还很模糊,希望接下来能把它弄得更懂一点,先记个板子
2601 调皮的小Biu(后缀数组,倍增算法,计算最长公共字串,板子)_第1张图片
2601 调皮的小Biu(后缀数组,倍增算法,计算最长公共字串,板子)_第2张图片
在这里插入图片描述

#include
using namespace std;
const int N=1e5*2+50;
int n,nn,k;
int rnk[N],tmp[N],sa[N],high[N],str[N]; 
bool cmp(int i,int j)
{
 if(rnk[i]!=rnk[j]) return rnk[i];
 int ri=(i+k<=n?rnk[i+k]:-1);
 int rj=(j+k<=n?rnk[j+k]:-1);
 return ri;
}
void get_sa()//使用倍增算法计算名次数组rank和后缀数组sa 
{
 for(int i=0;i<=n;i++)//初始化 
 {
  sa[i]=i;
  rnk[i]=(i:-1);//结尾以-1标记 
 }
 for(k=1;k<=n;k*=2)//倍增 
 {
  sort(sa,sa+n+1,cmp);
  tmp[sa[0]]=0;
  for(int i=1;i<=n;i++)
   tmp[sa[i]]=tmp[sa[i-1]]+(cmp(sa[i-1],sa[i])?1:0);
  for(int i=0;i<=n;i++)
   rnk[i]=tmp[i];
 } 
}
void get_high()//求后缀数组的最长公共前缀的长度
{
 /*for(int i=0;i<=n;i++)//rank与sa互为逆 
  rnk[sa[i]]=i;*/
 int h=0;
 high[0]=0;
 for(int i=0;i;i++)
 {
  int j=sa[rnk[i]-1];
  if(h>0)
   h--;
  for(;j+h;h++)
   if(str[j+h]!=str[i+h])
    break;
  high[rnk[i]-1]=h;
 }
}
void get_commonfix()//计算最长公共字串
{
 int ans=0;
 for(int i=0;i;i++)
  if(sa[i]!=nn&&sa[i+1]!=nn&&sa[i])
  ans=max(ans,high[i]);
 cout<;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>nn;
    for(int i=0;i;i++)
     cin>>str[i];
    str[nn]=6;//中间间隔符
	for(int i=0;i;i++)
	 cin>>str[i+nn+1];
	n=nn*2+1;
	get_sa();//使用倍增算法计算名次数组rank和后缀数组sa 
	get_high();//求后缀数组的最长公共前缀的长度 
	get_commonfix();//计算最长公共字串
    return 0;
}

你可能感兴趣的:(2601 调皮的小Biu(后缀数组,倍增算法,计算最长公共字串,板子))