笔试题-根据词典中的词语，对文本进行分词

def get_res(lexicon, data):
    '''
    第一行是一个字符串数组，表示一组词库
    第二行是一个文本，表示要检查的文本内容
    想法：想划分最长的词语
    问题：有一些情况无法被考虑到
    '''
    # 80 %
    lexicon = sorted(lexicon, key=lambda x :len(x), reverse=False)
    # print(lexicon)
    cur_data = data[:]
    for word in lexicon:
        if word in cur_data:
            cur = cur_data.split(word)
            cur_data = "".join(cur)
    if len(cur_data) == 0:
        return "true"
    return "false"


def get_res2(lexicon, data):
    # 动态规划
    # dp[:i]表示前i字可以被分词成功
    n = len(data)
    dp = [False for _ in range( n +1)]
    dp[0] = True
    lexicon = set(lexicon)
    for i in range(1, n+ 1):
        if data[i - 1] in lexicon:
            dp[i] = True
        else:
            for j in range(i - 1, -1, -1):
                if dp[j]:
                    if data[j:i] in lexicon:
                        dp[i] = True

    if dp[n]:
        return "true"
    else:
        return "false"


if __name__ == "__main__":
    lexicon = input().split()
    data = input()
    # lexicon = ["my", "name", "is", "red"]
    # data = "ismynamered"
    res = get_res2(lexicon, data)
    print(res)