結果

問題 No.731 等差数列がだいすき
コンテスト
ユーザー rookzeno
提出日時 2018-09-08 23:03:17
言語 Python3
(3.14.3 + numpy 2.4.4 + scipy 1.17.1)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
AC  
実行時間 525 ms / 1,500 ms
コード長 410 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 560 ms
コンパイル使用メモリ 20,828 KB
実行使用メモリ 35,680 KB
最終ジャッジ日時 2026-05-28 01:37:12
合計ジャッジ時間 17,970 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import numpy as np
n = int(input())
a = list(map(int,input().split()))
su1 = (sum(a)*2)
b1 = n*2
d1 = (0+2*(n-1))*n//2
su2 = 0
b2 = 0
d2 = 0
for i in range(n):
  su2 += a[i] * (i)*2
  b2 += (i)*2
  d2 += ((i)**2)*2
kake = np.array([[b1,d1],[b2,d2]])
ans = np.array([su1,su2])
ans = np.dot(np.linalg.inv(kake),ans)
print(ans[0],ans[1])
kota = 0
for i in range(n):
  kota += (a[i]-ans[0]-ans[1]*i)**2
print(kota)
0