【蓝桥杯】算法提高 字符串匹配

问题描述

　　给出一个字符串和多行文字，在这些文字中找到字符串出现的那些行。你的程序还需支持大小写敏感选项：当选项打开时，表示同一个字母的大写和小写看作不同的字符；当选项关闭时，表示同一个字母的大写和小写看作相同的字符。

输入格式

　　输入的第一行包含一个字符串S，由大小写英文字母组成。
　　第二行包含一个数字，表示大小写敏感的选项，当数字为0时表示大小写不敏感，当数字为1时表示大小写敏感。
　　第三行包含一个整数n，表示给出的文字的行数。
　　接下来n行，每行包含一个字符串，字符串由大小写英文字母组成，不含空格和其他字符。

输出格式

　　输出多行，每行包含一个字符串，按出现的顺序依次给出那些包含了字符串S的行。

#include <stdio.h>
#include <cstring>
#include <string.h>
#include <iostream>
#define MOD 1000000007
#define MAX_COL 9
using namespace std;
int next[101];
void get(string s){
    next[0] = -1;
    int i=0,j=-1;
    while(i<s.length()){
        if(j<0||s[i]==s[j]){
            i++;
            j++;
            next[i]=j;
        }else{
            j = next[j];
        }
    }
}

int kmp(string p,string s)//p是模板串，s是子串
{
    get(s);
    int i=0,j=0;
    int n=p.length();
    int m=s.length();
    while(i<n&&j<m)
    {
        if(j==-1||p[i]==s[j])
        {
            i++;
            j++;
        }
        else
            j=next[j];
    }
    if(j>=m)//s串比较完毕
        return i-m+1;
    else
        return -1;
}
int main(){
    //freopen("in.txt","r",stdin);
    string s;
    cin>>s;
    int t ,n ;
    cin>>t>>n;
    if(t==1){
        for(int tt= 0;tt<n;tt++){
            string p;
            cin>>p;
            if(kmp(p,s)>=1)cout<<p<<endl;
        }
    }else{
        for(int tt= 0;tt<n;tt++){
            string p;
            cin>>p;
            string p1 = p;
            for(int ij = 0 ;ij<s.length();ij++){
                if(s[ij]>='a'&&s[ij]<='z'){
                    s[ij] = s[ij] -'a'+'A';
                }
            } 
            for(int ij = 0 ;ij<p1.length();ij++){
                if(p1[ij]>='a'&&p1[ij]<='z'){
                    p1[ij] = p1[ij] -'a'+'A';
                }
            } 
            if(kmp(p1,s)>=1)cout<<p<<endl;
        }
    }
    
    return 0;
}