E. Compress Words

Posted 2022-07-03 liulex

E. Compress Words

KMP

#include<bits/stdc++.h>
using namespace std;
int p[1000005];
int len;
void getNext(char *s)

    int i=0,j=-1;
    p[0]=-1;
    int n=strlen(s);
    for(i; i<n;)
    
        if(j==-1||s[i]==s[j])
        
            i++;
            j++;
            if(s[i]==s[j])
                p[i]=p[j];
            else p[i]=j;

        
        else
        
            j=p[j];

        

    
    //return j;

int KMP(char *s,char *t)

    int n=strlen(t);

    int i=max(0,len-n),j=0;
    getNext(t);
    for(i; i<len;)
    
        while(s[i]!=t[j])
        
            if(p[j]==0)
            
                j=0;
                break;
            
            else j=p[j];
        
        if(s[i]==t[j])
        
            j++;
        
        i++;
    
    return j;


char s[1000004];
char t[1000004];
int main()

    int T;
    scanf("%d",&T);
    scanf("%s",s);
    len=strlen(s);
    T--;
    for(int _=0;_<T;_++)
        scanf("%s",t);
        int c=KMP(s,t);
        //int n=strlen(s);

        int m=strlen(t);
        int i=len-c;
        for(int j=0;j<m;j++)
            s[i++]=t[j];
        s[i]=‘\0‘;
        len+=m-c;
    
    cout<<s<<‘\n‘;