C++实现CCF201409-3字符串匹配

问题描述

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

输入格式

　　输入的第一行包含一个字符串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<vector>

using namespace std;
int main()
{
	string s, str, str_copy; 
	vector<string> ans;
	int flag, n, index;
	cin>>s>>flag>>n;
	if(flag == 0) // 大小写不敏感，先将待匹配字符串全部转换为小写字母
		transform(s.begin(), s.end(), s.begin(), ::tolower);
	for(int i = 0; i < n; i++)
	{
		cin>>str;
		str_copy = str;  // 记录原始字符串，防止因为大小写不敏感而影响结果
		if(flag == 0)
		{
			transform(str.begin(), str.end(), str.begin(), ::tolower);
		}
		index = str.find(s); // 查找子串，若返回-1，则没有该子串
		if(index != -1)
			ans.push_back(str_copy);  // 注意记录str_copy结果
	}
	
	vector<string>::iterator iter;
	for(iter=ans.begin(); iter!=ans.end(); iter++)
		cout<<*iter<<endl;
	return 0;
}