bzoj 1398 寻找主人 Necklace

Description

给定两个项链的表示，判断他们是否可能是一条项链。

Input

输入文件只有两行，每行一个由0至9组成的字符串，描述一个项链的表示（保证项链的长度是相等的）。

Output

如果两条项链不可能同构，那么输出’No’，否则的话，第一行输出一个’Yes’

第二行输出该项链的字典序最小的表示。 设L = 项链长度，L <= 1000000。

Sample Input

2234342423
2423223434

Sample Output

Yes
2234342423

 

这题用最小表示法$O(n)$的做法，或者用后缀数组和后缀自动机都可以做。  

 

#include<bits/stdc++.h>
using namespace std; 
int const N=1000000+100;  
char s1[N],s2[N];   
int  solve(char *s){
    int len=strlen(s);  
    int i=0,j=1,k=0;  
    while (i<len && j<len && k<len){
        int t=s[(i+k)%len]-s[(j+k)%len]; 
        if(!t) k++;  
        else {
            if(t>0) i+=k+1; 
            else if(t<0) j+=k+1;  
            if(i==j) j++;  
            k=0; 
        } 
    } 
    return min(i,j);  
}      
         
int main(){
    scanf("%s",s1);  
    int t1=solve(s1); 
    scanf("%s",s2);  
    int t2=solve(s2);     
    int check=0; 
    int len1=strlen(s1); 
    int len2=strlen(s2);  
    if(len1!=len2){
        puts("NO");  
        return 0; 
    }  
    for(int i=0;i<len1;i++)  
        if( s1[(t1+i) % len1]!=s2[(t2+i)%len1] )  
            check=1;  
    if(check){
        puts("No");
        return 0; 
    }  
    puts("Yes");  
    for(int i=0;i<len1;i++)  
        putchar(s1[(t1+i)%len1]);  
    return 0;
} 

 后缀自动机写法，由于要超空间，所以使用了链表的方法来存储边。 虽然也可以过，但是比之前的方法慢了很多。  

#include<bits/stdc++.h>
using namespace std;  
int const N=1900000+10;  
int fa[N<<1],len[N<<1];  
struct Edge{
    int to,nt,w;     
}e[N<<1];  
char s[1000001];     
int ls,tot,cnt,h[N<<1],id,lk[N<<1];  
void inline addedge(int a,int b,int c){
    e[++cnt].to=b;  
    e[cnt].w=c;  
    e[cnt].nt=h[a];  
    h[a]=cnt;  
} 
int inline son(int x,int y){
    for(int i=h[x];i;i=e[i].nt)  
        if(e[i].w==y) {
            id=i;  
            return e[i].to; 
        }       
    return 0;       
}  
void add(int c){
    int p=ls;  
    int np=ls=++tot;  
    len[np]=len[p]+1; 
    for(;p &&!son(p,c);p=fa[p]) addedge(p,np,c);        
    if(!p) fa[np]=1;  
    else {
        int q=son(p,c);  
        if(len[q]==len[p]+1) fa[np]=q;    
        else {
            int nq=++tot;  
            fa[nq]=fa[q];  
            for(int i=h[q];i;i=e[i].nt)  
                addedge(nq,e[i].to,e[i].w);  
            len[nq]=len[p]+1;   
            fa[q]=fa[np]=nq;  
            for(;p && son(p,c)==q;p=fa[p])  
                e[id].to=nq;  
        }  
    } 
}  
void solve(char *s){
    int p=1;  
    for(int i=0;s[i];i++){
        int c=s[i]-48; 
        if(!son(p,c)) {
            puts("No"); 
            exit(0);  
        }  
        p=son(p,c);    
    }  
}  
int main(){
    scanf("%s",s);  
    int len1=strlen(s);
    tot=ls=1;    
    for(int i=0;s[i];i++)  
        add(s[i]-48);
    for(int i=0;s[i];i++)  
        add(s[i]-48);  
    scanf("%s",s); 
    int len2=strlen(s); 
    if(len1!=len2){
        puts("No");
        return 0; 
    }  
    solve(s);   
    vector<int> num(len1+2<<1);  
    vector<int> sa(len1+2<<2);  
    for(int i=1;i<=tot;i++)  num[len[i]]++;  
    for(int i=1;i<=2*len1;i++) num[i]+=num[i-1];  
    for(int i=1;i<=tot;i++)  sa[num[len[i]]--]=i;   
    num.clear();  
    sa.clear();  
    for(int i=tot;i>=1;i--){
        int t=sa[i];  
        int f=fa[t]; 
        for(int j=h[t];j;j=e[j].nt){
            int v=e[j].to;  
            if(!v) continue;  
            lk[t]=max(lk[t],lk[v]+1);  
        } 
    }  
    puts("Yes");  
    int p=1;  
    for(int i=0;i<len1;i++){
        for(int j=0;j<10;j++){
            int t=son(p,j);          
            if(!t) continue;  
            if(lk[t]+i+1<len1)  continue;  
            putchar(48+j);  
            p=t;   
            break;  
        } 
    } 
    return 0 ; 
}