L1-064 估值一亿的AI核心代码 (20 分)
本题要求你实现一个稍微更值钱一点的 AI 英文问答程序,规则是:
- 无论用户说什么,首先把对方说的话在一行中原样打印出来;
- 消除原文中多余空格:把相邻单词间的多个空格换成 1个空格,把行首尾的空格全部删掉,把标点符号前面的空格删掉;
- 把原文中所有大写英文字母变成小写,除了 I;
- 把原文中所有独立的
can you
、could you
对应地换成I can
、I could
—— 这里“独立”是指被空格或标点符号分隔开的单词; 把原文中所有独立的 I
和me
换成you
; 把原文中所有的问号?
换成惊叹号!
; 在一行中输出替换后的句子作为 AI 的回答。
输入格式:
输入首先在第一行给出不超过 10 的正整数 N,随后 N 行,每行给出一句不超过 1000 个字符的、以回车结尾的用户的对话,对话为非空字符串,仅包括字母、数字、空格、可见的半角标点符号。
输出格式:
按题面要求输出,每个 AI 的回答前要加上 AI:
和一个空格。
输入样例:
6
Hello ?
Good to chat with you
can you speak Chinese?
Really?
Could you show me
What Is this prime? I,don 't know
输出样例:
1. List item
Hello ?
AI: hello!
Good to chat with you
AI: good to chat with you
can you speak Chinese?
AI: I can speak chinese!
Really?
AI: really!
Could you show me 5
AI: I could show you 5
What Is this prime? I,don 't know
AI: what Is this prime! you,don't know
先上代码:
import re # 开心
if __name__ == '__main__':
number = int(input())
questions = []
answers = []
for i in range(number):
mes = input()
questions.append(mes) # 保存原始字符串
reg = re.compile(r'\s')
mes = reg.sub(r' ', mes) # 把所有分隔符换成空格
mes = mes.split(' ') # 用空格切片
mes = [item for item in mes if item != ''] # 消除空格
mes = ' '.join(mes) # 重新用空格连接字符串
reg = re.compile(r' +(\W)') # 消除标点符号前的空格 -- 测试点 0,2
mes = reg.sub(r"\1", mes)
mes = mes.replace('I', '$') # 保护大写I
mes = mes.lower() # 全部变小写, 包括Me -- 测试点 1
mes = mes.replace('$', 'I') # 还原大写I
reg = re.compile(r'^(I|me)(?= |\W)|(?<= |\W)(I|me)(?= |\W)|(?<= |\W)(I|me)$|^(I|me)$')
mes = reg.sub(r'/*/', mes) # 替换独立的I或me, 用/*/代替, 避免影响接下来对can you和could you的替换
reg = re.compile(r'^can you(?= |\W)|(?<= |\W)can you(?= |\W)|(?<= |\W)can you$|^can you$')
mes = reg.sub(r'I can', mes) # 替换独立的can you
reg = re.compile(r'^could you(?= |\W)|(?<= |\W)could you(?= |\W)|(?<= |\W)could you$|^could you$')
mes = reg.sub(r'I could', mes)
reg = re.compile(r'^/\*/(?= |\W)|(?<= |\W)/\*/(?= |\W)|(?<= |\W)/\*/$|^/\*/$')
mes = reg.sub(r'you', mes) # 还原you
mes = mes.replace('?', '!') # 替换所有的?为!
answers.append('AI: ' + mes) # 在修改后字符串前加上'AI:'
for i in range(number):
print(questions[i] + '\n' + answers[i], end='')
if i != number - 1:
print()
单独讲一下正则表达式,题目的要求是:“独立”是指被空格或标点符号分隔开的单词
的,这其实有四种情况(测试点3, 4):
- 左边无任何字符,位于句首,右边被空格或者标点符号分隔。
^can you(?= |\W)
- 左右都被空格或者标点符号分隔。
(?<= |\W)can you(?= |\W)
- 右边无任何字符,位于句首,左边被空格或者标点符号分隔
(?<= |\W)can you$
- 左右都无字符,单独成句。
^can you$
上面Me
也要被换成you
(测试点1),所以要先小写化,再替换,每一步操作的顺序,都不能随意颠倒。
终于AC了。