codeforces 427D Match & Catch（后缀数组，字符串）

题目

 

参考：http://blog.csdn.net/xiefubao/article/details/24934617

 

题意：给两个字符串，求一个最短的子串。使得这个子串在两个字符串中出现的次数都等于1.出现的定义为：可以重叠的出现。

 

解法：后缀数组的应用。从小枚举长度。如果一个长度len合法的话：则一定存在这个样的sa[i]排名。sa[i]与s[i+1]的公共前缀长度大于等于len，且sa[i]与[i-1]的公共前缀长度小于len，同时sa[i+1]与[i+2]的公共前缀长度小于len，同时保证sa[i]与sa[i+1]在两个串中。Judge函数就是技巧性地实现了这些判断。

 

 

//如果一个长度len合法的话：则一定存在这个样的sa[i]排名。
//sa[i]与s[i+1]的公共前缀长度大于等于len，
//且sa[i]与[i-1]的公共前缀长度小于len，
//同时 sa[i+1]与[i+2]的公共前缀长度小于len，
//同时保证sa[i]与sa[i+1]在两个串中。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxn 1000001
int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b]&&r[a+l]==r[b+l];}
void da(int *r,int *sa,int n,int m){
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=0;i<m;i++) ws[i]=0;
     for(i=0;i<n;i++) ws[x[i]=r[i]]++;
     for(i=1;i<m;i++) ws[i]+=ws[i-1];
     for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
     for(j=1,p=1;p<n;j*=2,m=p){
       for(p=0,i=n-j;i<n;i++) y[p++]=i;
       for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
       for(i=0;i<n;i++) wv[i]=x[y[i]];
       for(i=0;i<m;i++) ws[i]=0;
       for(i=0;i<n;i++) ws[wv[i]]++;
       for(i=1;i<m;i++) ws[i]+=ws[i-1];
       for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
       for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
       x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
     return;
}
int rankk[maxn],height[maxn];
void calheight(int *r,int *sa,int n){
     int i,j,k=0;
     for(i=1;i<=n;i++) rankk[sa[i]]=i;
     for(i=0;i<n;height[rankk[i++]]=k)
     for(k?k--:0,j=sa[rankk[i]-1];r[i+k]==r[j+k];k++);
     return;
}

char s[10010];
int len1,len2;
int r[10010],sa[10010];
//这是一个除去下标影响的好方法呀
int judge(int n,int k ){
    int a,b;a=b=0;
    for(int i=2;i<=n;i++){
        if(height[i]<k){
            if(a==1&&b==1)return 1;
            a=0,b=0;
        }
        if(sa[i]>=0&&sa[i]<len1)a++;
        if(sa[i]>len1&&sa[i]<len2)b++;    
    }
    return 0;
}

int main(){
    scanf("%s",s);
    len1=strlen(s);
    s[len1]=1;
    scanf("%s",s+len1+1);
    len2=strlen(s);
    for(int i=0;i<len2;i++)r[i]=s[i];
    r[len2]=0;
    da(r,sa,len2+1,128);
    calheight(r,sa,len2);
    int ans=-1;
    for(int i=1;i<=len1;i++){//枚举长度
        if(judge(len2+1,i)){
            ans=i;break;
        }
    }
    printf("%d\n",ans);
    return 0;
}