爱奇艺 2021秋招在线笔试

题目描述

一个已经用分词算法分好词的中文句子，由于分词算法有一定的错误率使得某些词汇没有正确分词，所以希望用一个词典中的词去进行匹配并把句中所有能完全匹配上的字符串强行改成一个词，但其它不受影响的分词结果不改变，请实现满足这个要求的算法。（注：测试数据中不出现中文字符，均使用类似样例2的字符，且不会出现多解，匹配词相互有冲突的情况）

输入描述

第一行是分好词的一句话（字符串），词与词间由空格分开；第二行是若干个需要匹配的词，词与词间有空格分开

输出描述

修改后的分词结果（一个字符串），词与词间由空格分开

示例1

样例输入

aa bcd edf deda
ded

样例输出

aa bc ded f ded a

Python 代码

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# 类似于序列化反序列化，先把空格应该放的位置记录下来，再复写
def main():
    words = list(map(str, input().split())) # word是原来的按空格分割的list
    s = ''.join(words) # s是去空格
    pattern = list(map(str, input().split()))# pattern是字典
    temp = []
    l = 0
    # tmp存放空格的位置
    for i in range(len(words)):
        temp.append(l + len(words[i]))
        l += len(words[i])

    for x in pattern:# x是字典中的值
        i = 0
        cnt = 0
        while i < len(s):
            if x[cnt] == s[i]:
                cnt += 1
            elif cnt > 0:
                cnt = 0

            if cnt == len(x):
                temp.append(i + 1)#把空格的位置记录下来
                temp.append(i + 1 - cnt)
                cnt = 0
                break

            i += 1

    ret = []
    pre = 0
    temp.sort()
	# 反序列化
    for x in temp:
        if pre != x:
            ret.append(s[pre:x])
            ret.append(" ")
            pre = x

    print(''.join(ret))

if __name__ == '__main__':
    main()