题解 | #矩阵乘法计算量估算#
矩阵乘法计算量估算
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()