結果
問題 | No.2447 行列累乗根 |
ユーザー |
![]() |
提出日時 | 2023-08-25 23:12:29 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 973 bytes |
コンパイル時間 | 363 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 97,956 KB |
最終ジャッジ日時 | 2024-12-24 11:00:52 |
合計ジャッジ時間 | 40,666 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 20 TLE * 8 |
ソースコード
from sys import stdin import numpy as np input = stdin.readline def cb(a): if a >= 0: return pow(a, 1 / 3) else: return -pow(-a, 1 / 3) T = int(input()) ret: list[str] = [] for _ in range(T): a, b = map(float, input().split()) c, d = map(float, input().split()) A = np.array([[a, b], [c, d]]) eigval, eigvec = np.linalg.eigh(A) M = np.array([[eigvec[0][0], eigvec[0][1]], [eigvec[1][0], eigvec[1][1]]]) e0 = cb(eigval[0]) e1 = cb(eigval[1]) m0 = eigvec[0][0] m1 = eigvec[0][1] m2 = eigvec[1][0] m3 = eigvec[1][1] # B = M @ np.diag([e0, e1]) @ M.T # ret.append("{:1.5f} {:1.5f}".format(B[0][0], B[0][1])) # ret.append("{:1.5f} {:1.5f}".format(B[1][0], B[1][1])) ret.append("{:1.5f} {:1.5f}".format(m0 * m0 * e0 + m1 * m1 * e1, m0 * m2 * e0 + m1 * m3 * e1)) ret.append("{:1.5f} {:1.5f}".format(m2 * m0 * e0 + m3 * m1 * e1, m2 * m2 * e0 + m3 * m3 * e1)) print(*ret, sep="\n")