The Preliminary Contest for ICPC Asia Xuzhou 2019 D. Carneginon （KMP）

 D. Carneginon

題目鏈接：https://nanti.jisuanke.com/t/41386

題目大意：給一個母串T，接下來給n個串S，如果S的長度與T的長度相等：如果S與T串相同，輸出：jntm!，否則輸出：friend!， 如果S的長度比T小，如果S是T的子串，輸出：my child!，否則輸出：oh, child! ， 如果S的長度比T大，如果T是S的子串，輸出：my teacher!，否則輸出：senior!

思路：KMP判子串

代碼：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=100005;
char S[N],T[N];
int nex[N],nexx[N];
void getnext(int tlen)
{
    int i=-1,j=0;
    nex[0]=-1;
    while(j<tlen)
    {
        if(i==-1||T[i]==T[j]) nex[++j]=++i;
        else i=nex[i];
    }
}
void getnexx(int tlen)
{
    int i=-1,j=0;
    nexx[0]=-1;
    while(j<tlen)
    {
        if(i==-1||S[i]==S[j]) nexx[++j]=++i;
        else i=nexx[i];
    }
}
int KMP(int slen,int tlen,int *ne) //子串出現最早位置
{
    int i=0,j=0,ma=0;
    while(i<slen&&j<tlen)
    {
        if(j==-1||S[i]==T[j]) i++,j++;
        else j=ne[j];
    }
    if(j==tlen) return 1;
    return 0;
}

int main()
{
    int n;
    scanf("%s%d",T,&n);
    int tlen=strlen(T);
    getnext(tlen);
    for(int i=0;i<n;i++)
    {
        scanf("%s",S);
        int slen=strlen(S);
        if(tlen<slen)
        {
            if(KMP(slen,tlen,nex)) printf("my teacher!\n");
            else printf("senior!\n");
        }
        else if(tlen==slen)
        {
            if(KMP(slen,tlen,nex)) printf("jntm!\n");
            else printf("friend!\n");
        }
        else if(tlen>slen)
        {
            getnexx(slen);
            if(KMP(tlen,slen,nexx)) printf("my child!\n");
            else printf("oh, child!\n");
        }
    }
    return 0;
}