题解 | #矩阵乘法计算量估算#

矩阵乘法计算量估算

https://www.nowcoder.com/practice/15e41630514445719a942e004edc0a5b

通过字典存储每个字母的矩阵。

通过while一直循环查找')',找到时计算')'左边的两个字母,再把计算后字母的替换掉为新字母,同时更新字典里的字母的矩阵。

def fun():
    import string
    n = int(input())
    alpha_map_size = {i: list(map(int, input().split(' '))) for i in string.ascii_uppercase[0:n]}
    rule = input()
    res = 0
    while True:
        # 寻找要计算的字符串组合
        sub_rule = rule[rule.index(')') - 2:rule.index(')')]
        # 计算字符串
        a1 = alpha_map_size.get(sub_rule[0])
        a2 = alpha_map_size.get(sub_rule[1])
        sub_value, new_size, new_alpha = a1[0] * a1[1] * a2[1], [a1[0], a2[1]], sub_rule[0]
        # 更新信息
        res += sub_value
        rule, alpha_map_size[new_alpha] = rule.replace(rule[rule.index(')') - 3:rule.index(')') + 1], new_alpha), new_size
        if len(rule) == 1:
            break
    print(res)


fun()

全部评论

相关推荐

06-26 15:35
武汉大学 运营
点赞 评论 收藏
分享
真烦好烦真烦:豆包润色了自己没看看吗,再说了,都说豆包是愚蠢且勤快的大学生,ds才是聪明的研究生,怎么敢让豆包写论文的
点赞 评论 收藏
分享
明天不下雨了_人机版:让我们大声的说出来:以前的未来就是现在
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务