結果
問題 |
No.731 等差数列がだいすき
|
ユーザー |
|
提出日時 | 2018-09-17 14:16:25 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 460 ms / 1,500 ms |
コード長 | 434 bytes |
コンパイル時間 | 72 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 44,456 KB |
最終ジャッジ日時 | 2024-07-18 07:39:43 |
合計ジャッジ時間 | 10,867 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
import numpy as np n = int(input()) Y = [] meany = 0 for x in input().split(): x = int(x) meany += x Y.append(x) meany /= n meanx = (n - 1) / 2 Y = np.array(Y) - meany X = np.array(range(n)) - meanx # print(X) Sxy = 0 Sx2 = 0 for x, y in zip(X,Y): Sxy += x * y Sx2 += x ** 2 d = Sxy / Sx2 b = meany - d * meanx Y += meany ai = b cost = 0 for x in Y: cost += (x - ai) ** 2 ai += d print(b, d) print(cost)