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

矩阵乘法计算量估算

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

while True:
    try: 
        n = int(input())                        # 参与运算的矩阵个数
        diclist = []                           # 用来存储矩阵信息
        poplist = []                           # 存储出栈元素
        num = 0                       # 记录此时进栈元素的数量
        countsum = 0                            # 初始化乘法计算次数
        for i in range(n):
            diclist.append(list(map(int,input().split())))
        countlist  = list(input())

        for i,e in enumerate(countlist):
            if 'A' <= e <= 'Z':
                poplist.append(diclist[num])
                num += 1
            elif e == ')':
                # 计算每次的乘法次数
                countsum += poplist[-2][0] * poplist[-2][1] * poplist[-1][1]
                # 元素进行出栈
                a = poplist.pop()       
                b = poplist.pop()
                # 计算得到新矩阵后重新进栈
                poplist.append([b[0],a[1]])
        print(countsum)
    except:
        break

全部评论

相关推荐

09-19 13:59
门头沟学院 Java
用微笑面对困难:Trae一下,如果真成了,他用了直接发字节起诉代码版权,,这个代码不商用是没问题的如果没成也是情理之中的。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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