后缀数组专题

最新推荐文章于 2021-02-27 02:43:54 发布
最新推荐文章于 2021-02-27 02:43:54 发布
阅读量508 收藏 1
点赞数 1
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/mr_troy/article/details/69499578
版权

6道题题目链接：点击打开链接

（本篇为后缀数组的训练题的个人ac题解，非正规教学题解，仅供参考。）

（后缀数组在《算法竞赛入门经典训练指南》上有专门的题解，本篇代码部分完全采用此书上的代码作为模板）

题意理解：

A、求一个字符串中重复出现至少两次以上的但不重叠出现的最长子串的长度

B、求一个字符串中重复出现至少K次以上的但可重叠出现的最长子串的长度

C、求两个字符串中最长的公共子串的长度

D、求两个字符串中长度大于等于K的公共子串的总个数

E、求N个字符串的至少在半数以上的字符串中出现过的最长公共子串并输出，如果有多条则排序后输出

F、对于一个长度为N的字符串，切两刀成三段，每段翻转，使整体字典序最小，问两刀该切在哪里，并输出处理完后的字符串。

建议做题顺序：

1、F ，只需要用到sa数组

2、B ，先求sa数组，再求height数组，最后遍历一遍height数组（二分）

3、A ，先求sa数组，再求height数组，最后遍历一遍height数组（二分）

4、C ，将两个字符串拼接成一个字符串，然后求sa数组，再求height数组，最后遍历一遍height数组（二分）

5、E ，将N个字符串拼接成一个字符串，先二分求出最长长度，再按此长度找出符合的字符串输出（二分）

6、D （难）


详细题解

题目：F

题目描述：对于一个长度为N的字符串，切两刀成三段，每段翻转，使整体字典序最小，问两刀该切在哪里，并输出处理完后的字符串。

做法分析：

对于一个序列 10 1 2 3 4，切两刀，使整体最小，可以这么切 10 1 | 2 | 3 4 ，这样10 1逆序后为1 10 ； 2逆序为2 ； 3 4逆序为4 3。实际上这两刀我们可以一刀一刀切，我们可以看做第一刀为主要排序准则，第二刀为次要排序准则，第一刀切的好一定比第一刀没切好更优， 所以整体最优可以转化为局部最优，若想让第一段逆序后最小，可以看做将整个字符串逆序后构造后缀数组sa，那么sa【0】指的就是逆序后能最小的一段字符串的起始点，即sa【0】就是第一刀的位置。以上面的序列为例，将整体字符串反转后为4 3 2 1 10，那么sa【0】的位置在1的位置，因为1 10是最小的， 所以第一刀切在1前面。接下来要在4 3 2 中再切第二刀，所以对4 3 2这个序列再做一次后缀数组得到新的sa数组，由sa【0】得到第二刀的位置。

经过上面的分析，这道切两刀分三段的题就可以拓展到切N刀分N+1段的题。另外有点细节需要注意，因为至少要切2刀分3段，所以第一刀并不是一定是sa【0】，比如sa【0】的位置是这个序列的最后一个，那么第二刀就只能切空气了，所以要再加个判断sa【0】不行的话就sa【1】再不行sa【2】，找到第一刀的合理位置使第二刀有的切的位置。 


代码展示：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=200010;
int o[maxn],s[maxn],sa[maxn],t[maxn],t2[maxn],c[maxn];
struct node {
	int val,pos;
	bool operator < (const node & x) const {
		return val<x.val;
	}
} tmp[maxn];
void build_sa(int n,int m) {
	int i,*x=t,*y=t2;
	//初始化基数排序
	for (i=0; i<m; i++) c[i]=0;
	for (i=0; i<n; ++i) ++c[ x[i]=s[i] ];
	for (i=1; i<m; ++i) c[i]+=c[i-1];
	for (i=n-1; i>=0; --i) sa[ --c[x[i]] ]=i;
	//倍增基数排序
	for (int k=1; k<=n; k<<=1) {
		int p=0;
		//先根据sa数组直接排序第二关键词
		for (i=n-k; i<n; ++i) y[p++]=i;
		for (i=0; i<n; ++i) if (sa[i]>=k) y[p++]=sa[i]-k;
		//再排序第一关键词
		for (i=0; i<m; i++) c[i]=0;
		for (i=0; i<n; ++i) ++c[x[y[i]]];
		for (i=1; i<m; ++i) c[i]+=c[i-1];
		for (i=n-1; i>=0; --i) sa[--c[x[y[i]]] ]=y[i];
		//根据sa数组重新构造x数组
		swap(x,y);
		p=1;
		x[sa[0]]=0;
		for (i=1; i<n; ++i)
			x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
		if (p>=n) break;
		m=p;
	}
}
int main() {
	//freopen("datain.txt","r",stdin);
	ios::sync_with_stdio(false);
	cin.tie(0);
	//读入 
	int n;
	cin>>n;
	for (int i=0; i<n; ++i) {
		cin>>o[i]; 			//保留一开始的数值在o数组中 
		tmp[i].val=o[i];	//tmp数组用来排序 
		tmp[i].pos=i;
	}
	//排序并分配名次，同成绩的同一名次 
	sort(tmp,tmp+n);
	s[tmp[0].pos]=0;
	for (int i=1; i<n; ++i) {
		if (o[tmp[i].pos]==o[tmp[i-1].pos]) s[tmp[i].pos]=s[tmp[i-1].pos];
		else s[tmp[i].pos]=i;
	}
	//逆序 
	for (int i=0; i<=(n-1)/2; i++) swap(s[i],s[n-1-i]);
	//先找第一个切割点 
	build_sa(n,n);
	int first=0;
	while(sa[first]<=1) first++;
	int ans1=n-1-sa[first];
	first=sa[first];
	//再找第二个切割点 
	build_sa(first,n);
	int second=0;
	while(sa[second]<=0) second++;
	int ans2=n-1-sa[second];
	//输出 
	for (int i=ans1; i>=0; i--) cout<<o[i]<<endl;
	for (int i=ans2; i>ans1; i--) cout<<o[i]<<endl;
	for (int i=n-1; i>ans2; i--)  cout<<o[i]<<endl;
}

题目：B

题目描述：求一个字符串中重复出现至少K次以上的但可重叠出现的最长子串的长度

做法分析：

构造后缀数组sa，构造height数组，二分遍历长度L，每次分割height数组，保证每一段的height值不小于L，如果这一段中的字符串个数大于等于K，则当前L是符合条件的。

代码展示：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=20005;
int s[maxn];
int t[maxn],t2[maxn],sa[maxn],c[maxn];
void build_sa(int n,int m){
	int i,*x=t,*y=t2;
	//初始化基数排序 
	for (i=0;i<m;i++) c[i]=0;
	for (i=0;i<n;++i) ++c[ x[i]=s[i] ];
	for (i=1;i<m;++i) c[i]+=c[i-1];
	for (i=n-1;i>=0;--i) sa[ --c[x[i]] ]=i;
	//倍增基数排序 
	for (int k=1;k<=n;k<<=1){
		int p=0;
		//先根据sa数组直接排序第二关键词
		for (i=n-k;i<n;++i) y[p++]=i;
		for (i=0;i<n;++i) if (sa[i]>=k) y[p++]=sa[i]-k;
		//再排序第一关键词
		for (i=0;i<m;i++) c[i]=0;
		for (i=0;i<n;++i) ++c[x[y[i]]];
		for (i=1;i<m;++i) c[i]+=c[i-1];
		for (i=n-1;i>=0;--i) sa[--c[x[y[i]]] ]=y[i];
		//根据sa数组重新构造x数组
		swap(x,y);
		p=1;x[sa[0]]=0;
		for (i=1;i<n;++i)
			x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
		if (p>=n) break;
		m=p; 
	}
}
int rank[maxn],height[maxn];
void getHeight(int n){
	int k=0;
	for (int i=0;i<n;++i) rank[sa[i]]=i;
	for (int i=0;i<n;++i) {
		if (k) k--;
		if (rank[i]){
			int j=sa[rank[i]-1];
			while(s[i+k]==s[j+k]) k++;
		} 
		height[rank[i]]=k;
	}
}
bool judge(int n,int x,int k){
     int num=1,maxnum=1;
     for (int i=1;i<n;i++)
     {
         if (height[i]<x){
         	num=1;
        	continue;
         }
         num++;
         maxnum=max(maxnum,num);
         if (maxnum>=k) return true;
     }
     return false;    
}
int main(){
//	freopen("datain.txt","r",stdin);
	int n,k;
	while(~scanf("%d%d",&n,&k)){
		//读入 
		for (int i=0;i<n;++i) scanf("%d",&s[i]);
		//构造sa和height数组 
		build_sa(n,n);
		getHeight(n);
		//二分找最大的长度 
		int l=0,r=n,mid=1,ans=0;
		while(l<=r){
			mid=(l+r)/2;
			if (judge(n,mid,k)) {
				ans=mid;
				l=mid+1;
			}
			else r=mid-1;
		}
		printf("%d\n",ans);  
	}
}



题目：A

题目描述：求一个字符串中重复出现至少两次以上的但不重叠出现的最长子串的长度

做法分析：

构造后缀数组sa，构造height数组，二分遍历长度L，每次分割height数组，保证每一段的height值不小于L，如果这一段中的存在两个字符串不重叠，则当前L是符合条件的。需要注意的是题目中有一个特别要求，若字符串整个加上一个数等于另一个字符串，则也认为两个字符串是相等的，所以进行一次预处理

for (int i=0;i<n-1;++i) s[i]=s[i+1]-s[i]+89;

代码展示：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=20005;
int s[maxn];
int t[maxn],t2[maxn],sa[maxn],c[maxn];
void build_sa(int n,int m){
	int i,*x=t,*y=t2;
	//初始化基数排序 
	for (i=0;i<m;i++) c[i]=0;
	for (i=0;i<n;++i) ++c[ x[i]=s[i] ];
	for (i=1;i<m;++i) c[i]+=c[i-1];
	for (i=n-1;i>=0;--i) sa[ --c[x[i]] ]=i;
	//倍增基数排序 
	for (int k=1;k<=n;k<<=1){
		int p=0;
		//先根据sa数组直接排序第二关键词
		for (i=n-k;i<n;++i) y[p++]=i;
		for (i=0;i<n;++i) if (sa[i]>=k) y[p++]=sa[i]-k;
		//再排序第一关键词
		for (i=0;i<m;i++) c[i]=0;
		for (i=0;i<n;++i) ++c[x[y[i]]];
		for (i=1;i<m;++i) c[i]+=c[i-1];
		for (i=n-1;i>=0;--i) sa[--c[x[y[i]]] ]=y[i];
		//根据sa数组重新构造x数组
		swap(x,y);
		p=1;x[sa[0]]=0;
		for (i=1;i<n;++i)
			x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
		if (p>=n) break;
		m=p; 
	}
}
int rank[maxn],height[maxn];
void getHeight(int n){
	int k=0;
	for (int i=0;i<n;++i) rank[sa[i]]=i;
	for (int i=0;i<n;++i) {
		if (k) k--;
		if (rank[i]){
			int j=sa[rank[i]-1];
			while(s[i+k]==s[j+k]) k++;
		} 
		height[rank[i]]=k;
	}
}
bool judge(int n,int x){
     int mini=sa[0],maxi=sa[0];
     for (int i=1;i<n;i++)
     {
         if (height[i]<x){
            mini=sa[i];
            maxi=sa[i];
        	continue;
         }
         mini=min(mini,sa[i]);
         maxi=max(maxi,sa[i]);
         if (maxi-mini>x) return true;
     }
     return false;    
}
int main(){
//	freopen("datain.txt","r",stdin);
	int n;
	while(scanf("%d",&n)==1 && n){
		//读入 
		for (int i=0;i<n;++i) scanf("%d",&s[i]);
		//预处理 
		for (int i=0;i<n-1;++i) s[i]=s[i+1]-s[i]+89;
		//构造数组 
		build_sa(n,200);
		getHeight(n);
		//二分 
		int l=0,r=n,mid=1,ans=0;
		while(l<=r){
			mid=(l+r)/2;
			if (judge(n,mid)) {
				l=mid+1;
				ans=mid;
			}
			else r=mid-1;
		}  
		//输出 
		if (ans<4) printf("%d\n",0);
		else 	 printf("%d\n",ans+1);
	}
}


题目：C

题目描述：求两个字符串中最长的公共子串的长度

做法分析：

将两个字符串拼接成一个字符串，然后求sa数组，再求height数组，最后遍历一遍height数组（二分）

代码展示：

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=200010;
int s[maxn];
int t[maxn],t2[maxn],sa[maxn],c[maxn];
int len1;
void build_sa(int n,int m){
	int i,*x=t,*y=t2;
	//初始化基数排序 
	for (i=0;i<m;i++) c[i]=0;
	for (i=0;i<n;++i) ++c[ x[i]=s[i] ];
	for (i=1;i<m;++i) c[i]+=c[i-1];
	for (i=n-1;i>=0;--i) sa[ --c[x[i]] ]=i;
	//倍增基数排序 
	for (int k=1;k<=n;k<<=1){
		int p=0;
		//先根据sa数组直接排序第二关键词
		for (i=n-k;i<n;++i) y[p++]=i;
		for (i=0;i<n;++i) if (sa[i]>=k) y[p++]=sa[i]-k;
		//再排序第一关键词
		for (i=0;i<m;i++) c[i]=0;
		for (i=0;i<n;++i) ++c[x[y[i]]];
		for (i=1;i<m;++i) c[i]+=c[i-1];
		for (i=n-1;i>=0;--i) sa[--c[x[y[i]]] ]=y[i];
		//根据sa数组重新构造x数组
		swap(x,y);
		p=1;x[sa[0]]=0;
		for (i=1;i<n;++i)
			x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
		if (p>=n) break;
		m=p; 
	}
}
int rank[maxn],height[maxn];
void getHeight(int n){
	int k=0;
	for (int i=0;i<n;++i) rank[sa[i]]=i;
	for (int i=0;i<n;++i) {
		if (k) k--;
		if (rank[i]){
			int j=sa[rank[i]-1];
			while(s[i+k]==s[j+k]) k++;
		} 
		height[rank[i]]=k;
	}
}
bool judge(int n,int x){
     int l=sa[0],r=sa[0];
     for (int i=1;i<n;i++)
     {
         if (height[i]<x){
         	l=sa[i];
         	r=sa[i];
        	continue;
         }
         l=min(l,sa[i]);
         r=max(r,sa[i]); 
         if (l<len1 && r>=len1) return true;
     }
     return false;    
}
int main(){
//	freopen("datain.txt","r",stdin);
	ios::sync_with_stdio(false);
	cin.tie(0);
	//读入 
	string s1,s2;
	cin>>s1>>s2;
	int n=0;
	for (int i=0;s1[i];i++) s[n++]=s1[i]-'a'+2;
	len1=n;
	s[n++]=1;
	for (int i=0;s2[i];i++) s[n++]=s2[i]-'a'+2;
	//构造数组 
	build_sa(n,30);
	getHeight(n);
	//二分 
	int l=0,r=n,mid=1,ans=0;
	while(l<=r){
		mid=(l+r)/2;
		if (judge(n,mid)) {
			ans=mid;
			l=mid+1;
		}
		else r=mid-1;
	}  
	cout<<ans<<endl;
}


题目：E

题目描述：求N个字符串的至少在半数以上的字符串中出现过的最长公共子串并输出，如果有多条则排序后输出

做法分析：

将N个字符串拼接成一个字符串，每个串之间用一个从未出现过的字符分隔，然后求sa数组，再求height数组。

1、先二分求出至少在半数以上的字符串中出现过的最长的公共子串长度Lmax

2、再按长度为Lmax的情况下找符合条件的字符串依次输出

难点分析：判断height【i】中sa[i]和sa【i-1】所处的字符串若是同一个，则说明这是一个字符串自身的公共子串，这种情况不算做出现在一条字符串中；只有sa【i-1】和sa【i】分别处于两个字符串中才令个数增加。所以需要一个pos数组用来存储每一个i表示的字符原先是处于哪个字符串的。细节自行领悟，这里只做简要分析。

代码展示：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=200010;
int s[maxn]; 
int sa[maxn],t[maxn],t2[maxn],c[maxn],pos[maxn];
void build_sa(int n,int m){
	int i,*x=t,*y=t2;
	//初始化基数排序 
	for (i=0;i<m;i++) c[i]=0;
	for (i=0;i<n;++i) ++c[ x[i]=s[i] ];
	for (i=1;i<m;++i) c[i]+=c[i-1];
	for (i=n-1;i>=0;--i) sa[ --c[x[i]] ]=i;
	//倍增基数排序 
	for (int k=1;k<=n;k<<=1){
		int p=0;
		//先根据sa数组直接排序第二关键词
		for (i=n-k;i<n;++i) y[p++]=i;
		for (i=0;i<n;++i) if (sa[i]>=k) y[p++]=sa[i]-k;
		//再排序第一关键词
		for (i=0;i<m;i++) c[i]=0;
		for (i=0;i<n;++i) ++c[x[y[i]]];
		for (i=1;i<m;++i) c[i]+=c[i-1];
		for (i=n-1;i>=0;--i) sa[--c[x[y[i]]] ]=y[i];
		//根据sa数组重新构造x数组
		swap(x,y);
		p=1;x[sa[0]]=0;
		for (i=1;i<n;++i)
			x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
		if (p>=n) break;
		m=p; 
	}
}
int height[maxn],rank[maxn];
void getHeight(int n){
	int k=0;
	for (int i=0;i<n;++i) rank[sa[i]]=i;
	for (int i=0;i<n;++i) {
		if (k) k--;
		if (rank[i]){
			int j=sa[rank[i]-1];
			while(s[i+k]==s[j+k]) k++;
		} 
		height[rank[i]]=k;
	}
}
bool judge(int n,int l,int k){
	int num=0;
	bool vis[k];
	memset(vis,false,sizeof(vis));
	for (int i=1;i<n;++i){
		if (height[i]<l) {
			memset(vis,false,sizeof(vis));
			num=0;
		}
		else{
			if (pos[sa[i]]==pos[sa[i-1]]) continue;
			if (!vis[pos[sa[i]]]) {
				vis[pos[sa[i]]]=true;
				num++;
			}
			if (!vis[pos[sa[i-1]]]) {
				vis[pos[sa[i-1]]]=true;
				num++;
			}
			if (num > (k/2) ) return true;
		}
	}
	return false;
}
bool print(int n,int l,int k){
	int num=0;
	bool vis[k];
	memset(vis,false,sizeof(vis));
	for (int i=0;i<n;++i){
		if (height[i]<l) {
			if (num>(k/2)) {
				for (int j=0;j<l;j++) cout<<char(s[j+sa[i-1]]+'a');
				cout<<endl;
			}
			memset(vis,false,sizeof(vis)); 
			num=0;
		}
		else{
			if (pos[sa[i]]==pos[sa[i-1]]) continue;
			if (!vis[pos[sa[i]]]) {
				vis[pos[sa[i]]]=true;
				num++;
			}
			if (!vis[pos[sa[i-1]]]) {
				vis[pos[sa[i-1]]]=true;
				num++;
			}
		}
	}
	return false;
}
int main(){
//	freopen("datain.txt","r",stdin);
	ios::sync_with_stdio(false);
	cin.tie(0); 
	int k;
	while(cin>>k && k){
		//读入预处理，pos数组用来存储一开始每个字母所在的字符串编号 
		int n=0,tmp=0;
		for (int i=0;i<k;++i){
			string str;
			cin>>str;
			for (int j=0;str[j];++j) {
				pos[n]=tmp;
				s[n]=str[j]-'a';
				n++;
			}
			pos[n]=tmp;
			s[n]=30+(tmp++);
			n++; 
		} 
		//构造后缀数组 
		build_sa(n,30+tmp+5);
		getHeight(n);
		//二分找到出现在半数以上的字符串中的最长公共子串的长度ans 
		int l=0,r=n,ans=6;
		while(l<=r){
			int mid=(l+r)/2;
			if (judge(n,mid,k)) {
				ans=mid;
				l=mid+1;
			}
			else r=mid-1;
		} 
		//输出 
		if (ans) print(n,ans,k);
		else cout<<"?"<<endl;
		cout<<endl; 
	}
}


题目：D

题目描述：求两个字符串中长度大于等于K的公共子串的总个数

做法分析：

将两个个字符串拼接成一个字符串，串之间用一个从未出现过的字符分隔，然后求sa数组，再求height数组。

之后的做法中用到了单调栈的概念，我这里不具体拓展，如果想搞懂这道题先自行百度下单调栈的概念，掌握其代码实现。

顺序遍历一遍height数组，分段，使每段height值大于等于K，用一个单调栈维护当前状态，存储当前在第一个字符串的个数和在第二个字符串的个数。

这道题挺难理解的，我是死记的委屈，所以放在最后了。。，下面直接展示代码了，自行领悟吧，做了我好几天。。

代码展示：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=200010;
int s[maxn],sa[maxn],rank[maxn],t[maxn],t2[maxn],height[maxn],c[maxn];
int sta[maxn][2];
void build_sa(int n,int m){
	int i,*x=t,*y=t2;
	//初始化基数排序 
	for (i=0;i<m;i++) c[i]=0;
	for (i=0;i<n;++i) ++c[ x[i]=s[i] ];
	for (i=1;i<m;++i) c[i]+=c[i-1];
	for (i=n-1;i>=0;--i) sa[ --c[x[i]] ]=i;
	//倍增基数排序 
	for (int k=1;k<=n;k<<=1){
		int p=0;
		//先根据sa数组直接排序第二关键词
		for (i=n-k;i<n;++i) y[p++]=i;
		for (i=0;i<n;++i) if (sa[i]>=k) y[p++]=sa[i]-k;
		//再排序第一关键词
		for (i=0;i<m;i++) c[i]=0;
		for (i=0;i<n;++i) ++c[x[y[i]]];
		for (i=1;i<m;++i) c[i]+=c[i-1];
		for (i=n-1;i>=0;--i) sa[--c[x[y[i]]] ]=y[i];
		//根据sa数组重新构造x数组
		swap(x,y);
		p=1;x[sa[0]]=0;
		for (i=1;i<n;++i)
			x[sa[i]]= y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1:p++;
		if (p>=n) break;
		m=p; 
	}
}
void getHeight(int n){
	int k=0;
	for (int i=0;i<n;++i) rank[sa[i]]=i;
	for (int i=0;i<n;++i) {
		if (k) k--;
		if (rank[i]){
			int j=sa[rank[i]-1];
			while(s[i+k]==s[j+k]) k++;
		} 
		height[rank[i]]=k;
	}
}
int main(){
	//freopen("datain.txt","r",stdin);
	ios::sync_with_stdio(false);
	cin.tie(0); 
	int k;
	while(cin>>k && k){
		//读入 
		string s1,s2;
		cin>>s1>>s2;
		int len1=s1.length(),n=0;
		for (int i=0;s1[i];i++) s[n++]=s1[i]-'A'+2;
		s[n++]=1;
		for (int i=0;s2[i];i++) s[n++]=s2[i]-'A'+2;
		s[n++]=0;
		//构造数组 
		build_sa(n,70);
		getHeight(n);
		//预处理 
		for (int i=0;i<n;i++) height[i]=max(0,height[i]-k+1);	
		//sa[i-1]在第一个字符串中，sa[i]在第二个字符串中 
		long long ans=0,tot=0,top=0;
		for (int i=1;i<n;i++){
			if (height[i]<=0){
				top=0;tot=0;
			}else{
				int cnt=0;
				if (sa[i-1]<len1){ cnt++;tot+=height[i];
				}
				while(top && height[i]<=sta[top-1][0]){
					top--;
					tot+=(height[i]-sta[top][0])*sta[top][1];
					cnt+=sta[top][1];
				}
				sta[top][0]=height[i];
				sta[top++][1]=cnt;
				if (sa[i]>len1) ans+=tot; 
			}
		}
		//sa[i-1]在第二个字符串中，sa[i]在第一个字符串中 
		top=0;tot=0;
		for (int i=1;i<n;i++){
			if (height[i]<=0){
				top=0;tot=0;
			}else{
				int cnt=0;
				if (sa[i-1]>len1){ cnt++;tot+=height[i];
				}
				while(top && height[i]<=sta[top-1][0]){
					top--;
					tot+=(height[i]-sta[top][0])*sta[top][1];
					cnt+=sta[top][1];
				}
				sta[top][0]=height[i];
				sta[top++][1]=cnt;
				if (sa[i]<len1) ans+=tot; 
			}
		}
		//输出 
		cout<<ans<<endl;
	}
}

找到一个更好的后缀数组模版，作为通用模版

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=20000+100;
const int maxm=1000000+100;
struct SuffixArray {
    int s[maxn];
    int sa[maxn],rnk[maxn],height[maxn];
    int t1[maxn],t2[maxn],c[maxm],n;
    void build_sa(int m) {
        int i,*x=t1,*y=t2;
        for(i=0; i<m; i++) c[i]=0;
        for(i=0; i<n; i++) c[x[i]=s[i]]++;
        for(i=1; i<m; i++) c[i]+=c[i-1];
        for(i=n-1; i>=0; i--) sa[--c[x[i]]]=i;
        for(int k=1; k<=n; k<<=1) {
            int p=0;
            for(i=n-k; i<n; i++) y[p++]=i;
            for(i=0; i<n; i++)if(sa[i]>=k) y[p++]=sa[i]-k;
            for(i=0; i<m; i++) c[i]=0;
            for(i=0; i<n; i++) c[x[y[i]]]++;
            for(i=1; i<m; i++) c[i]+=c[i-1];
            for(i=n-1; i>=0; i--) sa[--c[x[y[i]]]]=y[i];
            swap(x,y);
            p=1;
            x[sa[0]]=0;
            for(i=1; i<n; i++)
                x[sa[i]]= y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]? p-1:p++;
            if(p>=n) break;
            m=p;
        }
    }
    void build_height() {
        int i,j,k=0;
        for(i=0; i<n; i++) rnk[sa[i]]=i;
        for(i=0; i<n; i++) {
            if(k)k--;
            j=sa[rnk[i]-1];
            while(s[i+k]==s[j+k]) k++;
            height[rnk[i]]=k;
        }
    }
    bool check(int l,int k) { //长l的出现k次以上的串
        int cnt=0;
        for(int i=1; i<n; i++) {
            if(height[i]<l)//新的一组开始
                cnt=1;
            else {
                cnt++;
                if(cnt>=k) return true;
            }
        }
        return false;
    }
    void print(int l,int k) {
        int cnt=0;
        for(int i=1; i<n; i++) {
            if(height[i]<l)//新的一组开始
                cnt=1;
            else {
                cnt++;
                if(cnt>=k) {
                    for (int j=0;j<l;j++) printf("%c",s[sa[i-1]+j]+'a'-1);
                    printf("\n");
                    return ;
                }
            }
        }
    }
} sa;
 
int main() {
    //freopen("datain.txt","r",stdin);
    string str;
    while(cin>>str) {
        for(int i=0; str[i]; i++) sa.s[i]=str[i]-'a'+1;
        sa.n=str.length()+1;
        sa.s[sa.n-1]=0;
        sa.build_sa(100+2);
        sa.build_height();
        int l=0,r=sa.n,maxlen,maxnum;
        while(l<=r) {
            int mid=(l+r)/2;
            if(sa.check(2,mid)) {
                l=mid+1;
                maxnum=mid;
            } else r=mid-1;
        }
        l=0;
        r=sa.n;
        while(l<=r) {
            int mid=(l+r)/2;
            if(sa.check(mid,maxnum)) {
                l=mid+1;
                maxlen=mid;
            } else r=mid-1;
        }
        sa.print(maxlen,maxnum);
        printf("%d\n",maxnum);
    }
    return 0;
}



Mr_Troy
关注
专栏目录
HDU - 3553:Just a String(后缀数组之求所有子串的第k大)
污miao汪的博客
07-24 672
  教练的专题，第一题模板题，第二题多校题，话说这题真的想了好久，后缀数组的用法真的太多了。。。 感觉很多时候都需要用二分来求解。特别这道题，感觉简直把二分运用到了极致。。。   题目大意： 给你一个串，求它所有子串的第k大（可重复）。   解题思路： 这道题目，首先我们要根据我们求出来的sa数组，因为我们sa数组是按字典序排好序的，所以理论上来说我们最小的子串应该是按照sa数组的...
树状数组专题 （差分思想)
点滴
08-16 551
树状数组专题 一直感觉树状数组用处比较小而且局限、因为最基本的用法就是单点更新和区间求和、但是线段树也能做，只不过代码长一点，但是仔细的去了解了一下树状数组以后发现有很多很赞的地方值得学习。 一、代码模板 1.单点更新、区间求和 for(int x = i; x <= n; x += x...
后缀数组专题训练
weixin_33893473的博客
01-05 95
后缀数组学习资料：http://blog.csdn.net/wxfwxf328/article/details/7599929 题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4691 思路：首先求出上下相邻的查询的字符串的LCP（最长公共前缀），这个可以通过求出的height数组得到。我们可以先求出上下相邻的字符串的rank，然后...
后缀数组专题整理
pcfbyssz的博客
08-21 328
先介绍一些基本概念： 子串 在字符串s中，取任意i&lt;=j，那么在s中截取从i到j的这一段就叫做s的一个子串   后缀  后缀就是从字符串的某个位置i到字符串末尾的子串，我们定义以s的第i个字符为第一个元素的后缀为suff(i)    后缀数组  把s的每个后缀按照字典序排序， 后缀数组sa[i]就表示排名为i的后缀的起始位置的下标 而它的映射数组rk[i]就表示起始位置的下标...
后缀数组专题小结
u011332631的专栏
07-15 981
关于后缀数组的话，我目前是只学了倍增算法，具体原理大白书讲的很清楚了，还有就是两篇集训队论文 国家集训队2004年论文《后缀数组》和国家集训队2009年论文《后缀数组——处理字符串的有力工具》。写的很清楚很详细ORZ 下面还是总结一下题目好了 【HDU 1403】 Longest Common Substring 这道题是后缀数组一道很基础的题了，将两个字符串连接在一起，中间再加上一个字符
[kuangbin带你飞]专题十八 后缀数组
Sega_hsj 's Blog
09-29 351
未完成 后缀数组
后缀数组java代码_后缀数组专题与代码模板
weixin_39535557的博客
02-27 132
因为 height[i] 表示 suffix( sa[i-1] ) 与 suffix( sa[i] )的 LCP， 我们统计出所有连续长度为K中的最小LCP，然后对所有最小LCP取最大值即为答案。这里可以用单调队列维护。#include#include#include#include#includeusing namespacestd;const int N = (int)2e5+10;int c...
20200212 专题：后缀自动机（SAM）
hongkongreporter
02-12 398
总览： OI WiKi（详细介绍） 后缀自动机可视化 概述： 能解决的问题： 检查字符串是否出现 不同子串个数 所有不同子串的总长度 字典序第 kk 大子串 最小循环移位 出现次数 第一次出现的位置 所有出现的位置 最短的没有出现的字符串 两个字符串的最长公共子串 多个字符串间的最长公共子串 时间和空间复杂度都是线性的！ 对于长度为nnn的字符串，最多2n−12n-12n−1个节点和3n−43n...
leetcode和oj-Search_Prefix_Suffix:搜索_前缀_后缀
06-30
"Search_Prefix_Suffix" 这个主题涉及到的是字符串处理中的一种常见问题，即查找字符串数组中的前缀或后缀。在实际的编程挑战中，这类问题经常出现，特别是在数据结构和算法的练习中。 首先，我们来理解一下什么是...
leetcode卡-LeetCode---Arrays-and-Strings:LeetCode---数组和字符串
07-06
8. **前缀和与后缀和**：前缀和是数组前n个元素的和，后缀和则是从某个位置到数组末尾的元素之和。这两个概念在解决区间和问题时非常有用。 9. **位操作**：在一些数组问题中，尤其是涉及整数的操作，位操作（如按...
配置中心.zip
最新发布
09-24
配置中心
基于java的物流管理系统报告的开题报告.docx
09-24
基于java的物流管理系统报告的开题报告.docx
企业级SpringCould脚手架工程：Eureka、Ribbon、Hystrix、Zuul、Feign、分布式事务.zip
09-24
企业级SpringCould脚手架工程：Eureka、Ribbon、Hystrix、Zuul、Feign、分布式事务、分布式锁、
基于Java的微信小程序html2wxml转换接口设计源码
09-24
该项目是基于Java的微信小程序html2wxml转换接口设计源码，共包含50个文件，其中包括10个属性配置文件、9个XML配置文件、6个首选项文件、6个Java源文件、3个Shell脚本、2个项目配置文件、2个HTML模板文件、1个类路径配置文件、1个Git忽略配置文件和1个JS类型定义文件。该解决方案利用JFinal、Jsoup和FastJson技术，为微信小程序提供高效的富文本渲染能力。
嵌入式学习之电子音乐门铃.zip
09-24
嵌入式学习资料电子信息与技术
习题集Python基础习题集.zip
09-24
【习题集】Python基础习题集.zip 【习题集】Python基础习题集.zip 【习题集】Python基础习题集.zip 【习题集】Python基础习题集.zip
基于Python、HTML、JavaScript和CSS的鹿鹿泽健身网站设计源码
09-24
该项目是一个基于Python、HTML、JavaScript和CSS的健身网站设计源码，名为鹿鹿泽健身。它包含94个文件，其中36个Python文件、28个HTML文件、5个VBS脚本、4个JavaScript文件、3个配置文件、3个SVG图形文件、3个PNG图片文件、2个文本文件、2个批处理文件、2个JPG图片文件。这个网站适用于健身爱好者，提供丰富的健身内容和互动功能。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值