結果
問題 |
No.731 等差数列がだいすき
|
ユーザー |
![]() |
提出日時 | 2019-04-04 16:53:48 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,032 ms / 1,500 ms |
コード長 | 308 bytes |
コンパイル時間 | 282 ms |
コンパイル使用メモリ | 12,288 KB |
実行使用メモリ | 77,884 KB |
最終ジャッジ日時 | 2024-12-26 00:13:21 |
合計ジャッジ時間 | 17,873 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
import numpy as np N = int(input()) y = list(map(int, input().split())) x = [i for i in range(N)] var_x = np.var(x) mu_x = np.mean(x) mu_y = np.mean(y) cov = sum((x-mu_x)*(y-mu_y)) / N a = cov / var_x b = mu_y - a*mu_x cost = 0 for i in range(N): cost += (y[i] - a*x[i] - b) ** 2 print(b, a) print(cost)