結果
問題 | No.2438 Double Least Square |
ユーザー | abap34 |
提出日時 | 2023-08-13 02:04:38 |
言語 | Julia (1.10.2) |
結果 |
AC
|
実行時間 | 1,908 ms / 2,000 ms |
コード長 | 2,591 bytes |
コンパイル時間 | 354 ms |
コンパイル使用メモリ | 7,200 KB |
実行使用メモリ | 340,208 KB |
最終ジャッジ日時 | 2024-10-01 17:31:07 |
合計ジャッジ時間 | 63,181 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,649 ms
308,660 KB |
testcase_01 | AC | 1,660 ms
308,268 KB |
testcase_02 | AC | 1,661 ms
308,076 KB |
testcase_03 | AC | 1,682 ms
307,952 KB |
testcase_04 | AC | 1,714 ms
309,076 KB |
testcase_05 | AC | 1,685 ms
308,920 KB |
testcase_06 | AC | 1,657 ms
308,784 KB |
testcase_07 | AC | 1,650 ms
308,732 KB |
testcase_08 | AC | 1,669 ms
308,948 KB |
testcase_09 | AC | 1,647 ms
308,204 KB |
testcase_10 | AC | 1,881 ms
340,196 KB |
testcase_11 | AC | 1,899 ms
335,076 KB |
testcase_12 | AC | 1,893 ms
335,116 KB |
testcase_13 | AC | 1,880 ms
338,868 KB |
testcase_14 | AC | 1,874 ms
337,636 KB |
testcase_15 | AC | 1,899 ms
337,384 KB |
testcase_16 | AC | 1,884 ms
339,460 KB |
testcase_17 | AC | 1,902 ms
337,968 KB |
testcase_18 | AC | 1,903 ms
335,712 KB |
testcase_19 | AC | 1,904 ms
339,888 KB |
testcase_20 | AC | 1,879 ms
337,276 KB |
testcase_21 | AC | 1,859 ms
337,212 KB |
testcase_22 | AC | 1,875 ms
337,624 KB |
testcase_23 | AC | 1,891 ms
339,388 KB |
testcase_24 | AC | 1,885 ms
340,208 KB |
testcase_25 | AC | 1,881 ms
336,004 KB |
testcase_26 | AC | 1,886 ms
337,072 KB |
testcase_27 | AC | 1,892 ms
337,328 KB |
testcase_28 | AC | 1,897 ms
335,088 KB |
testcase_29 | AC | 1,908 ms
339,876 KB |
testcase_30 | AC | 1,656 ms
307,884 KB |
testcase_31 | AC | 1,665 ms
308,244 KB |
testcase_32 | AC | 1,631 ms
308,624 KB |
ソースコード
using Printf function score(xy, x², y²) if x² == 0 return 0 end coef = xy / x² return y² + (coef^2 * x²) - (2coef * xy) end function main() N = parse(Int, readline()) H = parse(Int, readline()) x = zeros(Int, N) y = zeros(Int, N) for i in 1:N x[i], y[i] = parse.(Int, split(readline())) end INF = 10^16 slopes = zeros(Rational{Int}, N) for i in eachindex(x) if x[i] == 0 slopes[i] = INF else slopes[i] = (y[i] - (H // 2)) // x[i] end end slopes_perm = sortperm(slopes, rev=true) x_perm = sortperm(x, rev=true) # 全部lower xy = sum(x .* y) x² = sum(x .* x) y² = sum(y .* y) L_best = score(xy, x², y²) for i in eachindex(x) upper = slopes_perm[begin:i] lower = slopes_perm[i+1:end] upper_set = Set(upper) upper_x = getindex.(Ref(x), upper) upper_y = getindex.(Ref(y .- H), upper) upper_xy = sum(upper_x .* upper_y) upper_x² = sum(upper_x .* upper_x) upper_y² = sum(upper_y .* upper_y) lower_x = getindex.(Ref(x), lower) lower_y = getindex.(Ref(y), lower) lower_xy = sum(lower_x .* lower_y) lower_x² = sum(lower_x .* lower_x) lower_y² = sum(lower_y .* lower_y) L_u = score(upper_xy, upper_x², upper_y²) L_b = score(lower_xy, lower_x², lower_y²) L = L_u + L_b if L < L_best L_best = L end for j in eachindex(x) new_idx = x_perm[j] if new_idx in upper_set upper_xy -= x[new_idx] * (y[new_idx] - H) upper_x² -= x[new_idx]^2 upper_y² -= (y[new_idx] - H)^2 lower_xy += x[new_idx] * y[new_idx] lower_x² += x[new_idx]^2 lower_y² += y[new_idx]^2 else upper_xy += x[new_idx] * (y[new_idx] - H) upper_x² += x[new_idx]^2 upper_y² += (y[new_idx] - H)^2 lower_xy -= x[new_idx] * y[new_idx] lower_x² -= x[new_idx]^2 lower_y² -= y[new_idx]^2 end L_u = score(upper_xy, upper_x², upper_y²) L_b = score(lower_xy, lower_x², lower_y²) L = L_u + L_b if L < L_best L_best = L end end end @printf "%0.20f\n" L_best end main()