后缀数组+单调栈--poj3415 Common Substrings

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sizeof_you/article/details/84964122

传送门
时隔多年(其实也就几天 )我终于把这道题A了

就是用普通的后缀数组结合单调栈,加了个 > k >k 的限制,和[HAOI2016]找相同字符思路差不多,但有一些细节比如 h [ i ] h[i] 是和上一名的 l c p lcp ,所以有些判断要 1 -1 之类的

一直被多组数据坑··· s [ n + 1 ] s[n+1] 一定要置 0 0

代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define LL long long
#define maxn 400005
using namespace std;
int n,k,m,tax[maxn],rk[maxn],sa[maxn],h[maxn],st[maxn][20],tp[maxn];
int stk[maxn],top,f[maxn];
char s[maxn],a[maxn];

inline void rsort(){
    for(int i=1;i<=m;i++) tax[i]=0;
    for(int i=1;i<=n;i++) tax[rk[i]]++;
    for(int i=1;i<=m;i++) tax[i]+=tax[i-1];
    for(int i=n;i;i--) sa[tax[rk[tp[i]]]--]=tp[i];
}

inline void ssort(){
    for(int i=1;i<=n;i++) rk[i]=s[i],tp[i]=i;
    rsort();
    for(int w=1,p=0;w<=n&&p<n;w<<=1,m=p){
        p=0;
        for(int i=n-w+1;i<=n;i++) tp[++p]=i;
        for(int i=1;i<=n;i++)
            if(sa[i]>w) tp[++p]=sa[i]-w;
        rsort(); swap(rk,tp);
        rk[sa[1]]=p=1;
        for(int i=2;i<=n;i++)
            if(tp[sa[i]]==tp[sa[i-1]] && tp[min(n+1,sa[i]+w)]==tp[min(n+1,sa[i-1]+w)])
                rk[sa[i]]=p;
            else rk[sa[i]]=++p;
    }
}

inline void get_h(){
    int j,k=0;
    for(int i=1;i<=n;i++){
        if(k) --k;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) ++k;
        h[rk[i]]=k;
    }
}

inline void prework(){
    for(int i=1;i<=n;i++) st[i][0]=h[i];
    for(int j=1;(1<<j)<n;j++)
        for(int i=1;i+(1<<j)-1<=n;j++)
            st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}

inline int query(int l,int r){
    if(l>r) swap(l,r);++l;
    int t=log2(r-l+1); return min(st[l][t],st[r-(1<<t)+1][t]);
}

int main(){
    while(~scanf("%d",&k) && k!=0){
        scanf("%s",s+1); m=127; n=strlen(s+1);
        scanf("%s",a+1); int nn=strlen(a+1);
        s[++n]='z'+1;
        for(int i=1;i<=nn;i++) s[++n]=a[i]; s[n+1]=0;//!
        ssort(); get_h(); prework();
        memset(f,0,sizeof f);
        LL ans=0,now=0; top=0;
        for(int i=1;i<=n;i++){
        	if(h[i]<k) top=0,now=0;
        	else{
        		int tmp=0;
	            while(stk[top]>h[i] && top){
	            	now+=1LL*(h[i]-stk[top])*f[top];
	            	tmp+=f[top--];
	            }
	            if(sa[i-1]>n-nn) tmp++,now+=h[i]-k+1;//sa[i-1]
	            stk[++top]=h[i]; f[top]=tmp;
	            if(sa[i]<n-nn) ans+=now;
			}
        }
        top=0,now=0;
        for(int i=1;i<=n;i++){
            if(h[i]<k) top=0,now=0;
            else{
            	int tmp=0;
	            while(stk[top]>h[i] && top){
	                now+=1LL*f[top]*(h[i]-stk[top]);
	                tmp+=f[top--];
	            }
	            if(sa[i-1]<n-nn) tmp++,now+=h[i]-k+1;
	            stk[++top]=h[i]; f[top]=tmp;
	            if(sa[i]>n-nn) ans+=now;
			}
        }
        printf("%lld\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sizeof_you/article/details/84964122