poj3415

 

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<set>
#include<queue>
using namespace std;
#define ll long long
int n,m;
const int ding=2e5+5;
const int INF=0x3f3f3f3f;
int sa[ding],x[ding],y[ding],c[ding],strlenn[ding],ss[ding][2];
int r[ding],rank[ding],height[ding];
void get_SA(){
	int i,k;
	memset(c,0,sizeof(c));//清空桶;
	for(i=0;i<n;i++)	c[x[i]=r[i]]++; 
	for(i=1;i<m;i++)	c[i]+=c[i-1];
	for(i=n-1;i>=0;i--){
		sa[--c[x[i]]]=i;
	}
	int p=1;
	for(k=1;k<n;k<<=1,m=p){
		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;
		//大于等于k 
//sa是第一关键字的排名, 
//然后再根据第二关键字的排名来确定排名, 
		memset(c,0,sizeof(c));
		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数组	
			sa[--c[x[y[i]]]]=y[i];     
		memcpy(y,x,sizeof(x));
		memset(x,0,sizeof(x));
//		swap(x,y);
		p=1;//从1开始编号 
		x[sa[0]]=0;
//因为上面已经重置了x的值  所以下面是从1开始循环
//然后是和前一个比较sa[i]和sa[i-1] 比较 
		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++;
//x数组里面记住的是x[i]=j  下标为i的后缀,排名为j
//y数组的内容是y[i]=j 下标为j 的后缀排名为i 
		if(p>=n)break;
	}
}
void get_height(){
	int i,k=0,j;
	for(i=0;i<n;i++) rank[sa[i]]=i;
	for(i=0;i<n-1;i++){
//为什么循环到n-1,因为我们在原始字符串后面加了a
//所以这个后缀一定是最小的。 
//cout<<"2222222222"<<endl;
		if(k)k--;
		j=sa[rank[i]-1];
		for(;r[i+k]==r[j+k];k++);
		height[rank[i]]=k; 
	}	
} 
int main()
{
// 	std::ios::sync_with_stdio(false);
	int t,i,cnt;
	char str[ding],ch[ding];
	ll top,tot;
	while(scanf("%d",&t)!=EOF&&t)
	{//cout<<"t   "<<t<<endl;
		m=300;
		scanf("%s%s",str,ch);
		n=0;
		int len1=strlen(str);
		for(i=0;i<len1;i++)
		{
//			strlenn[n]=1;
			r[n++]=(int)str[i];
		}
//		strlenn[n]=0;
		r[n++]=1;//要用一个较小的数分割不知道为啥 
		int len2=strlen(ch);
		for(i=0;i<len2;i++)
		{
//			strlenn[n]=2;
			r[n++]=(int)ch[i];
		}
//		strlenn[n]=0;
		r[n++]=0;
		get_SA();
		get_height();
//		cout<<"qqqqq"<<endl;
		ll sum=0;
		top=tot=0;
		
		for(i=2;i<n;i++)
		{
			if(height[i]>=t)
			{
				int cnt=0;
				if(sa[i-1]>len1)cnt++,tot+=height[i]-t+1;
				while(top>0&&height[i]<=ss[top-1][0])
				{
					top--;
					tot-=ss[top][1]*(ss[top][0]-height[i]);
					cnt+=ss[top][1];
				}
				ss[top][0]=height[i];
				ss[top++][1]=cnt;
				if(sa[i]<len1)sum+=tot;
			}
			else tot=top=0;
		}
		tot=top=0;
		for(i=2;i<n;i++)
		{
			if(height[i]>=t)
			{
				int cnt=0;
				if(sa[i-1]<len1)cnt++,tot+=height[i]-t+1;
				while(top>0&&height[i]<=ss[top-1][0])
				{
					top--;
					tot-=ss[top][1]*(ss[top][0]-height[i]);
					cnt+=ss[top][1];
				}
				ss[top][0]=height[i];
				ss[top++][1]=cnt;
				if(sa[i]>len1)sum+=tot;
			}
			else tot=top=0;
		}
		
		printf("%I64d\n",sum);
	} 
	
 return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值