結果
問題 |
No.2438 Double Least Square
|
ユーザー |
|
提出日時 | 2023-08-18 22:15:54 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,083 bytes |
コンパイル時間 | 455 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 161,656 KB |
最終ジャッジ日時 | 2024-11-28 07:49:53 |
合計ジャッジ時間 | 64,440 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 WA * 1 |
other | AC * 9 RE * 1 TLE * 20 |
ソースコード
import random N = int(input()) H = int(input()) X = [list(map(int,input().split())) for i in range(N)] ans = 10**18 def calc(coef, xp, yp): a2 = yp/(xp+1e-9) a1 = (yp-H)/(xp+1e-9) return coef[0] + a1*coef[1] + (a1*a1)*coef[2] +\ a2*coef[3] + (a2*a2)*coef[4] for xp in range(501): for yp in range(501): coef = [0]*5 for x, y in X: if (x <= xp and y <= yp) or (x > xp and y > yp): coef[0] += y**2 coef[3] += - 2*x*y coef[4] += x**2 else: coef[0] += (y - H)**2 coef[1] += - 2*x*(y - H) coef[2] += x**2 ans = min(calc(coef, xp, yp), ans) ans = min(calc(coef, xp, yp+1), ans) ans = min(calc(coef, xp+1, yp), ans) ans = min(calc(coef, xp+1, yp+1), ans) xtest = H/(- coef[3]/(2*(coef[4]+1e-9)) + coef[1]/(2*(coef[2]+1e-9))) ytest = xtest * (- coef[3]/(2*(coef[4]+1e-9))) ans = min(calc(coef, min(max(xtest,xp),xp+1), min(max(ytest,yp),yp+1)), ans) print(ans)