试题 算法提高 字符串匹配

思路：为0时我这里全部转成小写对比，为1时就直接比较即可，详看代码。

资源限制
时间限制：1.0s   内存限制：256.0MB
问题描述
　　给出一个字符串和多行文字，在这些文字中找到字符串出现的那些行。
　　你的程序还需支持大小写敏感选项：当选项打开时，表示同一个字母的大写和小写看作不同的字符；
　　当选项关闭时，表示同一个字母的大写和小写看作相同的字符。
输入格式
　　输入的第一行包含一个字符串S，由大小写英文字母组成。
　　第二行包含一个数字，表示大小写敏感的选项，当数字为0时表示大小写不敏感，当数字为1时表示大小写敏感。
　　第三行包含一个整数n，表示给出的文字的行数。
　　接下来n行，每行包含一个字符串，字符串由大小写英文字母组成，不含空格和其他字符。
输出格式
　　输出多行，每行包含一个字符串，按出现的顺序依次给出那些包含了字符串S的行。
样例输入
Hello
1
5
HelloWorld
HiHiHelloHiHi
GrepIsAGreatTool
HELLO
HELLOisNOTHello
样例输出
HelloWorld
HiHiHelloHiHi
HELLOisNOTHello

代码：

def judge(pattern,s,num):
    a = []
    if num == 1:
        for i in s:
            if pattern in i:
                a.append(i)
    else:
        for i in s:
            if pattern.lower() in i.lower():
                a.append(i)
    return a

if __name__ == "__main__":
    s = input() # 模板字符串
    a = int(input()) # 0：大小写不敏感，1：大小写敏感
    n = int(input()) # 行数
    b = []
    while n > 0:
        b.append(input())
        n -= 1
    result = judge(s,b,a)
    for i in result:
        print(i)