Submission #46627128

Source Code Expand

n, m = gets.chomp.split.map(&:to_i)
toasts = gets.chomp.split.map(&:to_i).sort.reverse

two_toasts_plate_count = n - m
one_toast_plate_count = m - two_toasts_plate_count

ans = 0
ans += toasts[0...one_toast_plate_count].map { |v| v ** 2 }.sum if one_toast_plate_count.positive?

if two_toasts_plate_count.positive?
  target_toasts = toasts[one_toast_plate_count..]
  two_toasts_plate_count.times do
    current_max = target_toasts.shift
    current_min = target_toasts.pop

    ans += (current_max + current_min) ** 2
  end
end

puts ans

Submission Info

Submission Time
Task A - Toasts for Breakfast Party
User thatblue
Language Ruby (ruby 3.2.2)
Score 300
Code Size 558 Byte
Status AC
Exec Time 112 ms
Memory 36408 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 300 / 300
Status
AC × 3
AC × 25
Set Name Test Cases
Sample 01-sample-01.txt, 01-sample-02.txt, 01-sample-03.txt
All 01-sample-01.txt, 01-sample-02.txt, 01-sample-03.txt, 02-min-01.txt, 03-max-01.txt, 04-random-01.txt, 04-random-02.txt, 04-random-03.txt, 04-random-04.txt, 04-random-05.txt, 05-large-rand-01.txt, 05-large-rand-02.txt, 05-large-rand-03.txt, 05-large-rand-04.txt, 05-large-rand-05.txt, 05-large-rand-06.txt, 05-large-rand-07.txt, 05-large-rand-08.txt, 05-large-rand-09.txt, 05-large-rand-10.txt, 06-sukunai-01.txt, 06-sukunai-02.txt, 06-sukunai-03.txt, 06-sukunai-04.txt, 06-sukunai-05.txt
Case Name Status Exec Time Memory
01-sample-01.txt AC 42 ms 17228 KiB
01-sample-02.txt AC 42 ms 17096 KiB
01-sample-03.txt AC 42 ms 17048 KiB
02-min-01.txt AC 41 ms 17144 KiB
03-max-01.txt AC 91 ms 36408 KiB
04-random-01.txt AC 62 ms 22572 KiB
04-random-02.txt AC 76 ms 26684 KiB
04-random-03.txt AC 45 ms 18340 KiB
04-random-04.txt AC 101 ms 32404 KiB
04-random-05.txt AC 57 ms 20884 KiB
05-large-rand-01.txt AC 108 ms 35192 KiB
05-large-rand-02.txt AC 109 ms 35248 KiB
05-large-rand-03.txt AC 109 ms 35124 KiB
05-large-rand-04.txt AC 108 ms 34768 KiB
05-large-rand-05.txt AC 108 ms 34964 KiB
05-large-rand-06.txt AC 110 ms 35628 KiB
05-large-rand-07.txt AC 110 ms 35512 KiB
05-large-rand-08.txt AC 112 ms 35596 KiB
05-large-rand-09.txt AC 110 ms 34768 KiB
05-large-rand-10.txt AC 109 ms 34956 KiB
06-sukunai-01.txt AC 68 ms 26372 KiB
06-sukunai-02.txt AC 79 ms 28388 KiB
06-sukunai-03.txt AC 60 ms 22892 KiB
06-sukunai-04.txt AC 46 ms 18268 KiB
06-sukunai-05.txt AC 67 ms 24772 KiB