KMP进阶

终曲(加强版)

题目描述

最后的挑战终于到了！

站在yifenfei和MM面前的只剩下邪恶的大魔王lemon一人了！战胜他，yifenfei就能顺利救出MM。

Yifenfei和魔王lemon的挑战很简单：由lemon给出三个字符串，然后要yifenfei说出第一串的某个子串，要求该子串长度最小，并且同时包含第2个串和第3个串。

特别地，如果有多个这样的子串，则请输出字母序最小的一个。

输入描述

三行字符串，第一个字符串长度不超过2×1052×10​5​​

后面两个串的长度不超过100100。

输出描述

请对应每组输入数据输出满足条件的最短子串；

如果没有，请输出 “No”(注意大小写，不需要引号)

#include<bits/stdc++.h>
using namespace std;
int Next[200005];
vector<int> vl1,vl2,vr1,vr2;
int ansl,ansr=0x3f3f3f3f;
vector<pair<int,int> > rg;
string ans;
char a[200005];
void getans(string s){
    if(ans.size()==0||ans>s){
        ans=s;
    }
}
void update(int l,int r){
    if(ansr-ansl>r-l){
        ansr=r,ansl=l;
        rg.clear();
        rg.push_back(make_pair(l,r));
    }
    else if(ansr-ansl==r-l){
        rg.push_back(make_pair(l,r));
    }
}
void getNext(string s){
    Next[1]=0;
    for(int i=2,j=0;s[i];i++){
        while(j&&s[i]!=s[j+1]){
            j=Next[j];
        }
        if(s[i]==s[j+1]){
            j++;
        }
        Next[i]=j;
    }
}
void KMP(string s1,string s2,vector<int> &vl,vector<int> &vr){
    getNext(s2);
    for(int i=1,j=0;s1[i];i++){
        while(j&&s1[i]!=s2[j+1]){
            j=Next[j];
        }
        if(s1[i]==s2[j+1]){
            j++;
        }
        if(!s2[j+1]){
            vl.push_back(i-j+1);
            vr.push_back(i);
            j=Next[j];
        }
    }
}
int main(){
    string s1,s2,s3;
    cin>>s1;
    cin>>s2;
    cin>>s3;
    s1=" "+s1;
    s2=" "+s2;
    s3=" "+s3;
    KMP(s1,s2,vl1,vr1);
    KMP(s1,s3,vl2,vr2);
    if(vl1.size()==0||vl2.size()==0){
        cout<<"No";
        return 0;
    }
    int n=vl1.size(),m=vl2.size();
    for(int i=0;i<n;i++){
        int idx=lower_bound(vr2.begin(),vr2.end(),vl1[i])-vr2.begin();
        if(idx!=m){
            update(min(vl1[i],vl2[idx]),max(vr1[i],vr2[idx]));
        }
        idx--;
        if(idx>=0){
            update(min(vl1[i],vl2[idx]),max(vr1[i],vr2[idx]));
        }
//        idx=lower_bound(vl2.begin(),vl2.end(),vl1[i])-vl2.begin();
//        if(idx!=m){
//            update(min(vl1[i],vl2[idx]),max(vr1[i],vr2[idx]));
//        }
//        idx=lower_bound(vr2.begin(),vr2.end(),vr1[i])-vr2.begin();
//        if(idx!=m){
//            update(min(vl1[i],vl2[idx]),max(vr1[i],vr2[idx]));
//        }
    }
    for(int i=1;i<s1.size();i++){
        a[i]=s1[i];
    }
    for(int i=0;i<rg.size();i++){
        char t=a[rg[i].second+1];
        a[rg[i].second+1]='\0';
        getans(string(a+rg[i].first));
        a[rg[i].second+1]=t;
    }
    cout<<ans;
    return 0;
}

无线传输

题目描述

给你一个字符串 s​1​​，它是由某个字符串 s​2​​ 不断自我连接形成的（保证至少重复 2 次）。但是字符串 s​2​​ 是不确定的，现在只想知道它的最短长度是多少。

输入格式

第一行一个整数 L，表示给出字符串的长度。

第二行给出字符串 s​1​​ 的一个子串，全由小写字母组成。

输出格式

仅一行，表示 s​2​​ 的最短长度。

#include<bits/stdc++.h>
using namespace std;
int Next[1000005];
void getNext(string s){
    Next[1]=0;
    for(int i=2,j=0;s[i];i++){
        while(j&&s[i]!=s[j+1]){
            j=Next[j];
        }
        if(s[i]==s[j+1]){
            j++;
        }
        Next[i]=j;
    }
}
int main(){
    int n;
    string s;
    cin>>n;
    cin>>s;
    s=" "+s;
    getNext(s);
    cout<<s.size()-1-Next[s.size()-1];
    return 0;
}