CCF201409-3 字符串匹配 初级KMP 一句判断大小写锁定

问题描述

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

输入格式

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

输出格式

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

样例输入

Hello
1
5
HelloWorld
HiHiHelloHiHi
GrepIsAGreatTool
HELLO
HELLOisNOTHello

样例输出

HelloWorld
HiHiHelloHiHi
HELLOisNOTHello

样例说明

　　在上面的样例中，第四个字符串虽然也是Hello，但是大小写不正确。如果将输入的第二行改为0，则第四个字符串应该输出。

评测用例规模与约定

　　1<=n<=100，每个字符串的长度不超过100。

#include<iostream>#include<algorithm>#include<string>using namespace std;#define MAX 100+5const int cap = 'a' - 'A';int nx[MAX];void pre(string s){nx[0] = -1;int j = 0, k = -1;while (j < s.length()){if ((k == -1) || (s[j] == s[k])){j++;k++;nx[j] = k;}else k = nx[k];}}bool kmp(string s,string cs,int c){int z = 0;for (int i = 0; i <= s.length(); i++){if (z >= cs.length()){return true;}if (i == s.length()) return false;if (s[i] == cs[z] || abs(s[i]-cs[z])==cap*c){z++;}else{while (nx[z] != -1 && (s[i] != cs[z] && abs(s[i] - cs[z]) != cap*c)){ // 大小写z = nx[z];}}}}int main(){string s;cin >> s;pre(s);int c,n;cin >> c >> n;while (n--){string ss;cin >> ss;if (kmp(ss, s, abs(c-1))){cout << ss << endl;}}return 0;}