結果

問題 No.813 ユキちゃんの冒険
ユーザー tktk_snsn
提出日時 2020-12-08 21:30:38
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 783 ms / 2,000 ms
コード長 493 bytes
コンパイル時間 92 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 97,384 KB
最終ジャッジ日時 2024-09-21 08:35:05
合計ジャッジ時間 16,948 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
eps = 1e-6

n = int(input())
p = float(input())
q = float(input())
if abs(p - 1) < eps:
    print(1.0)
    exit()

A = np.eye(n * 2, dtype=np.float64)
X = np.arange(1, n * 2)
Y = np.arange(1, n * 2)
Y[X % 2 == 0] += 1
Y[X % 2 == 1] -= 1
np.add.at(A, (X, Y), -p)

X = np.arange(1, n * 2 - 1)
Z = np.arange(1, n * 2 - 1)
Z[X % 2 == 0] -= 2
Z[X % 2 == 1] += 2
np.add.at(A, (X, Z), -q)

b = np.zeros(2 * n, dtype=np.float64)
b[0] = 1.0

res = np.linalg.solve(A, b)
print(res[1])
0