結果

問題 No.731 等差数列がだいすき
ユーザー kohei2019
提出日時 2022-03-27 10:39:22
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 760 bytes
コンパイル時間 350 ms
コンパイル使用メモリ 82,240 KB
実行使用メモリ 76,536 KB
最終ジャッジ日時 2024-11-06 05:43:36
合計ジャッジ時間 17,906 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 15 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff #

N = int(input())
lsa = list(map(int,input().split()))

def calc_r(p):#p=(x,y)
    r = 0
    x,y = p
    for i in range(N):
        r += abs(i*x+y-lsa[i])**2
    return r


def calc_y(x):
    low_y = -10**5
    high_y = 10**5
    cnt = 200
    while cnt:
        cnt -= 1
        c1 = (low_y*2+high_y)/3
        c2 = (low_y+high_y*2)/3
        if calc_r((x,c1)) < calc_r((x,c2)):
            high_y = c2
        else:
            low_y = c1
    return low_y

low_x = -10**5
high_x = 10**5
cnt = 200
while cnt:
    cnt -= 1
    c1 = (low_x * 2 + high_x) / 3
    c2 = (low_x + high_x * 2) / 3
    if calc_r((c1,calc_y(c1))) < calc_r((c2,calc_y(c2))):
        high_x = c2
    else:
        low_x = c1
print(calc_y(low_x),low_x)
print(calc_r((low_x,calc_y(low_x))))
0