题解 | 三项评分线性定价

def read(t=int):
    return list(map(t,input().split()))

K=int(input())
line2=read()
N=int(input())
line4=read()

trains=[]
ys=[]
tests=[]
for i in range(K):
    trains.append([line2[i*4],line2[i*4+1],line2[i*4+2]])
    ys.append(line2[i*4+3])

for i in range(N):
    tests.append([line4[i*3],line4[i*3+1],line4[i*3+2]])

import numpy as np

X=np.array([[1]+e for e in trains])
Y=np.array(ys)
Xt=np.array([[1]+e for e in tests])

# print(X)
# print("---")
# print(Y)
# print("---")
# print(Xt)


W= np.linalg.inv(X.T @ X) @ X.T @ Y
Yt= Xt @ W

tests_y=Yt.tolist()

print(" ".join([str(round(e)) for e in tests_y]))

线性代数