結果

問題 No.731 等差数列がだいすき
ユーザー kohei2019kohei2019
提出日時 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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 92 ms
75,924 KB
testcase_01 AC 96 ms
76,048 KB
testcase_02 AC 98 ms
76,220 KB
testcase_03 AC 367 ms
76,320 KB
testcase_04 AC 847 ms
76,172 KB
testcase_05 AC 365 ms
76,256 KB
testcase_06 AC 1,177 ms
76,368 KB
testcase_07 AC 695 ms
75,916 KB
testcase_08 AC 837 ms
76,080 KB
testcase_09 AC 1,025 ms
76,316 KB
testcase_10 AC 1,224 ms
76,372 KB
testcase_11 AC 118 ms
76,104 KB
testcase_12 AC 598 ms
76,024 KB
testcase_13 AC 790 ms
75,980 KB
testcase_14 AC 920 ms
76,112 KB
testcase_15 WA -
testcase_16 AC 629 ms
76,112 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 1,373 ms
75,952 KB
testcase_20 AC 1,378 ms
76,376 KB
権限があれば一括ダウンロードができます

ソースコード

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