結果

問題 No.2447 行列累乗根
コンテスト
ユーザー hitonanode
提出日時 2023-08-25 23:05:18
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 629 bytes
コンパイル時間 292 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 109,396 KB
最終ジャッジ日時 2024-12-24 10:47:16
合計ジャッジ時間 52,022 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import numpy as np
from scipy.linalg import eigh

def cb(a):
    if a >= 0:
        return pow(a, 1 / 3)
    else:
        return -pow(-a, 1 / 3)

T = int(input())


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])

    B = M @ np.diag([e0, e1]) @ M.T
    print("{:1.5f}".format(B[0][0]), "{:1.5f}".format(B[0][1]))
    print("{:1.5f}".format(B[1][0]), "{:1.5f}".format(B[1][1]))
0