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

矩阵乘法计算量估算

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()

全部评论

相关推荐

10-25 00:32
香梨想要offer:感觉考研以后好好学 后面能乱杀,目前这简历有点难
点赞 评论 收藏
分享
无敌虾孝子:喜欢爸爸还是喜欢妈妈
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务