ccf-csp 202305-2 矩阵运算 python

题目

样例输入

3 2
1 2
3 4
5 6
10 10
-20 -20
30 30
6 5
4 3
2 1
4 0 -5

样例输出

480 240
0 0
-2200 -1100

代码

n,d=map(int,input().split(' '))#输入3 2
#map() 会根据提供的函数对指定序列做映射
#n=int(n)
#d=int(d)
​
#n行d列
#QKV
Q=[]
K=[]
V=[]
W=[]
for i in range(n):#######退回缩进ctrl+[
    
    Q.append(list(map(int,input().split(' '))))
for i in range(n):#######退回缩进ctrl+[
    
    K.append(list(map(int,input().split(' '))))
for i in range(n):#######退回缩进ctrl+[
    
    V.append(list(map(int,input().split(' '))))
W=(list(map(int,input().split(' '))))
​
##计算转置
KT=[]
for t in range(d):
    temp=[]
    for tt in K:
        temp.append(int(tt[t]))
    KT.append(temp)
​
def ss(left, right, a, b, c):
    # left:a*b right:b*c return ans
    ans = [[0]*c for i in range(a)]
    for i in range(a):
        for j in range(c):
            # get ans[i][j]
            temp = 0
            for k in range(b):
                temp += left[i][k]*right[k][j]
            ans[i][j]=temp
    return ans
​
​
​
#矩阵相乘
#KT*V
asn=ss(KT,V,d,n,d)
result=ss(Q,asn,n,d,d)
​
​
for i in range (n):
    for j in range (d):
        result[i][j]*=W[i]
for i in range(n):
    for j in range(d):
        print(result[i][j], end=' ')
    print()
​
​

注意

代码优化

1先算KT X V得到d X d 的矩阵，再算Q X KT 得到n X d的矩阵

2矩阵计算用函数可以节省大量内存！！！！

3map() 会根据提供的函数对指定序列做映射

矩阵计算代码

def ss(left, right, a, b, c):#a行b列 left X  b行c列 right
    ans = [[0]*c for i in range(a)]
    for i in range(a):
        for j in range(c):
            # get ans[i][j]
            temp = 0
            for k in range(b):
                temp += left[i][k]*right[k][j]
            ans[i][j]=temp
    return ans

print(result[i][j], end=' ')# 以空格为分隔符，不换行，但是最后有空格