結果
問題 |
No.731 等差数列がだいすき
|
ユーザー |
![]() |
提出日時 | 2020-01-02 15:30:04 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 516 ms / 1,500 ms |
コード長 | 434 bytes |
コンパイル時間 | 177 ms |
コンパイル使用メモリ | 12,416 KB |
実行使用メモリ | 44,588 KB |
最終ジャッジ日時 | 2024-11-22 18:06:00 |
合計ジャッジ時間 | 12,148 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np # いわゆる最小二乗法 N = int(readline()) x = np.arange(N) y = np.array(read().split(),np.int64) mu_x = x.mean(); mu_y = y.mean() a = np.dot(x - mu_x, y - mu_y) / np.dot(x - mu_x, x - mu_x) fit_y = a * (x - mu_x) + mu_y error = ((y - fit_y) ** 2).sum() print(fit_y[0], a) print(error)