E - Text Editor 二分,kmp

 

E - Text Editor

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
//string a,b,t;
char a[maxn],b[maxn],t[maxn];
int n;
int nxt[maxn];
int lena;
int lenb;
int cnt;
void getnxt()
{
    nxt[1]=0;
    for(int i=2,j=0; i<=lenb; i++)
    {
        while(j>0&&b[i]!=b[j+1]) j=nxt[j];
        if(b[i]==b[j+1]) j++;
        nxt[i]=j;
    }
}

void kmp()
{ 
    for(int i=1,j=0; i<=lena; i++)
    {
        while(j>0&&(j==lenb||a[i]!=b[j+1])) j=nxt[j];
        if(a[i]==b[j+1]) j++;
        if(j==lenb) cnt++;
    }
}

int check(int mid)
{
    cnt=0;
    lenb=mid;
    kmp();
    if(cnt>=n) return 1;
    else return 0;
}
int main()
{
    gets(a+1);
    gets(b+1);
    scanf("%d",&n);
    lena=strlen(a+1);
    lenb=strlen(b+1);
    int l=0,r=lenb;
    getnxt();
    while(l<r)
    {
        int mid=(l+r+1)/2;
        if(check(mid))
            l=mid;
        else
            r=mid-1;
    }
    if(l==0) printf("IMPOSSIBLE");
    else
    {
        for(int i=1; i<=l; i++)
            printf("%c",b[i]);
    }
}