結果

問題 No.731 等差数列がだいすき
ユーザー kohei2019kohei2019
提出日時 2022-03-27 10:38:19
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 760 bytes
コンパイル時間 376 ms
コンパイル使用メモリ 82,132 KB
実行使用メモリ 76,416 KB
最終ジャッジ日時 2024-11-06 05:42:03
合計ジャッジ時間 6,588 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 67 ms
76,040 KB
testcase_01 AC 71 ms
75,984 KB
testcase_02 AC 71 ms
76,036 KB
testcase_03 AC 143 ms
76,036 KB
testcase_04 AC 271 ms
76,052 KB
testcase_05 AC 147 ms
76,384 KB
testcase_06 AC 352 ms
76,048 KB
testcase_07 AC 229 ms
76,020 KB
testcase_08 AC 265 ms
76,076 KB
testcase_09 AC 313 ms
76,024 KB
testcase_10 AC 362 ms
76,148 KB
testcase_11 AC 80 ms
76,416 KB
testcase_12 AC 204 ms
76,384 KB
testcase_13 AC 254 ms
75,972 KB
testcase_14 AC 284 ms
76,112 KB
testcase_15 WA -
testcase_16 AC 213 ms
76,152 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 406 ms
76,164 KB
testcase_20 AC 404 ms
75,960 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 = 100
    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 = 100
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