結果
問題 |
No.813 ユキちゃんの冒険
|
ユーザー |
![]() |
提出日時 | 2020-03-20 09:46:50 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 782 ms / 2,000 ms |
コード長 | 644 bytes |
コンパイル時間 | 134 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 101,712 KB |
最終ジャッジ日時 | 2024-09-21 08:34:05 |
合計ジャッジ時間 | 18,285 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
#!/usr/bin/ python3.8 # %% import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np N, p, q = read().split() N = int(N) p, q = map(float, (p, q)) if q == 0: print(0) exit() elif p == 0: print(1) exit() N += 1 A = np.zeros((N + N + 2, N + N + 2), np.float64) np.fill_diagonal(A, 1.0) np.fill_diagonal(A[1:N, 2: N + 1], -q) np.fill_diagonal(A[N + 2:N + N + 1, N + 1:N + N], -q) np.fill_diagonal(A[1:N, N + 1: N + N], -p) np.fill_diagonal(A[N + 2:N + N + 1, 2: N + 1], -p) y = np.zeros(N + N + 2) y[N + 1] = 1 x = np.linalg.solve(A, y) print(x[1])