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

