题目描述
矩阵乘法的运算量与矩阵乘法的顺序强相关。
例如:
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