python3练习-华为机试

描述

矩阵乘法的运算量与矩阵乘法的顺序强相关。
例如：

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

计算A*B*C有两种顺序：((AB)C)或者(A(BC))，前者需要计算15000次乘法，后者只需要3500次。

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

数据范围：矩阵个数：1≤�≤15 1≤n≤15 ，行列数：1≤����,����≤100 1≤rowi​,coli​≤100 ，保证给出的字符串表示的计算顺序唯一。

进阶：时间复杂度：�(�) O(n) ，空间复杂度：�(�) O(n) 

输入描述：

输入多行，先输入要计算乘法的矩阵个数n，每个矩阵的行数，列数，总共2n的数，最后输入要计算的法则
计算的法则为一个字符串，仅由左右括号和大写字母（'A'~'Z'）组成，保证括号是匹配的且输入合法！

输出描述：

输出需要进行的乘法次数

示例1

输入：

3
50 10
10 20
20 5
(A(BC))

输出：

3500

是不是看到一脸懵？嘿嘿高数都还给老师了是不~

没事，下面一张图带你梦回高数

图来自【机试题(实现语言：python3)】矩阵乘法计算量估算---列表_3L_csdn的博客-CSDN博客

 总的来说，知道怎么算后已经成功一半了！

思路：

        1.接收数据，存入字典；（字典的value可以是列表是我没想到的）

        2.处理数据：遍历规则，括号优先运算；

        矩阵A：m行，n列
        矩阵B：a行，b列
        AxB =C
        矩阵C的行=矩阵A的行
        矩阵C的列=矩阵B的列
        矩阵C的乘法运算次数 = 矩阵A的行 X 矩阵B的列  X     矩阵A的列
                                    多少                  单元

        3.输出；

num = int(input())
dict_a = {}    # 存放A[0]是多少，B[1]是多少的字典
for i in range(num):
    a_list = list(map(int, input().split()))   # 这个map函数好评
    dict_a[i] = a_list
role = input()  # 规则
# print(role)
# print(dict_a)   # {0: [50, 10], 1: [10, 20], 2: [20, 5]}
role_list = []   # 标记
count = 0     # 计算乘积
for j in role[1:-1]:   # 去头去尾
    if j == "(":
        continue
    elif j in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":
        role_list.append(ord(j) - 65)     # ord()求ACII码
    elif j == ")":
        F = role_list.pop()     # 取符号前的两个字母对应的下标
        E = role_list.pop()
        count += dict_a[E][0]*dict_a[E][1]*dict_a[F][1]   # 先计算括号里面的乘积
        dict_a[E] = [dict_a[E][0], dict_a[F][1]]    # 存计算后的矩阵
        del dict_a[F]
        role_list.append(E)   # 这步别忘了存回来
# print(role_list)
# print(dict_a) 
for s in range(len(role_list)-1):     # 按顺序计算乘积
    count += dict_a[role_list[s]][0]*dict_a[role_list[s]][1]*dict_a[role_list[s+1]][1]
    dict_a[role_list[s+1]] = [dict_a[role_list[s]][0], dict_a[role_list[s+1]][1]]
    del dict_a[role_list[s]]
print(count)    # congratulations!大功告成