70 矩阵乘法计算量估算

100 篇文章 1 订阅
4 篇文章 0 订阅

题目描述
矩阵乘法的运算量与矩阵乘法的顺序强相关。

例如:

A是一个50×10的矩阵,B是10×20的矩阵,C是20×5的矩阵

计算ABC有两种顺序:((AB)C)或者(A(BC)),前者需要计算15000次乘法,后者只需要3500次。

编写程序计算不同的计算顺序需要进行的乘法次数

输入描述:
输入多行,先输入要计算乘法的矩阵个数n,每个矩阵的行数,列数,总共2n的数,最后输入要计算的法则

输出描述:
输出需要进行的乘法次数

示例1
输入
3
50 10
10 20
20 5
(A(BC))
输出
3500
思路:1.将每个矩阵的【行,列】作为总的行列matrixarray的子列表
2.利用栈存矩阵大小和运算后的矩阵大小
3. 对规则遍历:遇到“)”出两个栈,分别为相乘矩阵的matrix2,和matrix1,得到相乘之后的【行,列】入栈
4. 如果没有括号,只是字母,则取对应的列表入栈

while True:
    try:
        n=int(input())
        matrixarray=[]
        stack=[]
        result=0
        for i in range(n):
            eachmatrix=[int(i) for i in input().split()]#将每个矩阵的行列值作为matrixarray的子列表
            matrixarray.append(eachmatrix)
        rule=input()
        for i in range(n*3-2):#n个矩阵输入的规则有3n-2个字串
            if rule[i]=='(':
                pass
            elif rule[i]==')':
                matrix2=stack.pop()#每个为两个矩阵相乘,先弹出相乘的第二个矩阵
                matrix1=stack.pop()
                result+=matrix1[0]*matrix2[0]*matrix2[1]
                stack.append([matrix1[0],matrix2[1]])#相乘之后矩阵的【行,列】加入stack
            else:#若规则中为字母,则添加到stack栈中
                stack.append(matrixarray[ord(rule[i])-65])#A/B/C矩阵对应matrixarray的第几个子列表
        print(result)
    except:
        break
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值